Pandas Row Select, где начинается строка с любым элементом в списке

Question

Я хочу выбрать подмножество строк в кадре данных pandas на основе определенного строкового столбца, где значение начинается с любого количества значений в списке.

Небольшая версия этого:

df = pd.DataFrame({'a': ['aa10', 'aa11', 'bb13', 'cc14']})
valids = ['aa', 'bb']

Итак, я хочу, чтобы в этом случае строки a начинались с aa или bb.

Answer 1

Вам нужно startswith

df.a.str.startswith(tuple(valids))
Out[191]: 
0     True
1     True
2     True
3    False
Name: a, dtype: bool

После фильтра с оригинальным df

df[df.a.str.startswith(tuple(valids))]
Out[192]: 
      a
0  aa10
1  aa11
2  bb13