python - 为列表中的元素过滤 Pandas DataFrame

<分区>

我有一个包含值和附加信息的 pandas DataFrame。我希望能够提取仅属于一种信息的值。我事先不知道将查询哪些值和多少值。因此，有可能只调用一次带有附加信息“foo”的值，有时带有附加信息“bar”和“baz”，因此使用简化的 DataFrame

import pandas as pd
df = pd.DataFrame(
    [[1, 'foo'], [2, 'bar'], [3, 'baz']], columns=['value', 'id'])

我试过了

result = df[df.id in ['foo', 'bar']]

但我只是得到一个 ValueError:Series 的真值不明确。使用 a.empty、a.bool()、a.item()、a.any() 或 a.all()。但是我无法获取 any()-Function 来给我结果...。

最佳答案

使用isin根据传入的列表测试列的成员资格:

In [30]:
df[df['id'].isin(['foo','bar'])]

Out[30]:
   value   id
0      1  foo
1      2  bar

这里 isin 生成一个 bool 掩码，我们用它来过滤 df:

In [31]:    
df['id'].isin(['foo','bar'])

Out[31]:
0     True
1     True
2    False
Name: id, dtype: bool

关于python - 为列表中的元素过滤 Pandas DataFrame，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/32618812/