Python поиск множества элементов в кадре данных - PullRequest
0 голосов
/ 25 марта 2020

У меня есть датафрейм, как показано ниже

df = 
     A      B 
0   foo    10
1   goo    20
2   hoo    30
3   foo    40
4   goo    50
3   hoo    60
4   goo    70

Я хочу найти, существует ли ['foo', 'hoo'] в df ['A'], и я хочу создать новый фрейм данных, используя эти строки.

My code: 
idx = []
for i in ['foo','hoo']:
    idx.append(df[df['A']==i].index)
newdf = df.index[idx]

Есть ли лучший способ сделать это?

1 Ответ

0 голосов
/ 25 марта 2020

Это может сработать? -

new_df = df[(df['A'] == 'foo') | (df['A'] == 'hoo')].reset_index(drop=True)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...