python - 通过从现有数据框 python 中选择特定行来创建新数据框

Question

我的 pandas 数据框中有一张表。 df

id count price
1    2     100
2    7      25
3    3     720
4    7     221
5    8     212
6    2     200

我想从中创建一个新的数据框 (df2)，选择计数为 2 且价格为 100 的行，计数为 7 且价格为 221

我的输出应该是 df2 =

id count price
1    2     100
4    7     221

我正在尝试使用 df[df['count'] == '2' & df['price'] == '100']

但出现错误

TypeError: cannot compare a dtyped [object] array with a scalar of type [bool]

Answer 1

你需要添加 () 因为 & 比 == 有更高的优先级:

df3 = df[(df['count'] == '2') & (df['price'] == '100')]
print (df3)
  id count price
0  1     2   100

如果需要检查多个值，请使用 isin :

df4 = df[(df['count'].isin(['2','7'])) & (df['price'].isin(['100', '221']))]
print (df4)
  id count price
0  1     2   100
3  4     7   221

但是如果检查数字，使用:

df3 = df[(df['count'] == 2) & (df['price'] == 100)]
print (df3)

df4 = df[(df['count'].isin([2,7])) & (df['price'].isin([100, 221]))]
print (df4)