标签: python pandas
我可以使用布尔索引选择子集:
df.loc[df['my column'] == 1]
我要做的是将== 1替换为列表[1, 2]。
== 1
[1, 2]
类似的东西:
df.loc[df['my column'] in [1,2]]
这相当于长或陈述。
我如何很好地使用熊猫?
(另外,非布尔答案也很好。)
答案 0 :(得分:3)
使用isin测试值列表中的成员资格:
isin
UITableView
示例:
df.loc[df['my column'].isin([1,2])]