Question

我有一个pandas数据帧，df，其中包含一个功能（'alpha'），它是一个字母列表{'A'，'B'，...，'G'}

我想从df中选择属于此功能子集的所有行，例如{'A'，'B'，'C'}。

最“pythonic”的方法是什么？

我正在思考以下几点：

subset = {'A','B','C'}
df1 = df[df['alpha'] == subset]

...但这会产生错误：

"need more than 0 values to unpack"

Answer 1

我认为你想使用isin来测试会员资格，例如：

In [79]:
subset = {'a','b','c'}
df = pd.DataFrame({'a':list('abasbvggcgasgfdasgcdce')})
df[df['a'].isin(subset)]

Out[79]:
    a
0   a
1   b
2   a
4   b
8   c
10  a
15  a
18  c
20  c

在python中选择值的子集

1 个答案: