У меня есть датафрейм, как показано ниже
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]
Есть ли лучший способ сделать это?