python - 如何在 Pandas 中返回具有多个条件的数据框

Question

这是示例数据框。

data = [['United Kingdom', 'High income'], ['Albania', 'Upper middle income'], ['Russia', 'Upper middle income'], ['Afganistan','Low income'], ['USA','High income']]
df = pd.DataFrame(data, columns = ['Country', 'Income Group']) 

在这里，我试图只返回高收入和中高收入的国家:

df = np.where(df['Income Group'] == 'High income' & df['Income Group'] == 'Upper middle income')

这是输出:

TypeError: tuple indices must be integers or slices, not str

但是，如果您对其他列使用相同的方法，则效果很好:

df = np.where(df['Country'] == 'USA')

“收入组”列有什么问题？

非常感谢任何帮助

Answer 1

isin在这里也很有用:

df[df['Income Group'].isin(['High income', 'Upper middle income'])]

输出:

          Country         Income Group
0  United Kingdom          High income
1         Albania  Upper middle income
2          Russia  Upper middle income
4             USA          High income