首先创建一个DataFrame:
fruits_dict = { 'Fruits':['Apple','Banana','Cherry','Dates','Eggfruit'],
'Quantity': [5, 10, 8, 3, 7],
'Color': ['Red', 'Yellow', 'Red', 'Brown', 'Yellow']}
df=pd.DataFrame(fruits_dict)
观察这个DataFrame:
Fruits Quantity Color
0 Apple 5 Red
1 Banana 10 Yellow
2 Cherry 8 Red
3 Dates 3 Brown
4 Eggfruit 7 Yellow
使用isin
进行判断:
cond=[3,5,7]
df=df[df['Quantity'].isin(cond)]
结果:
Fruits Quantity Color
0 Apple 5 Red
3 Dates 3 Brown
4 Eggfruit 7 Yellow