使用值列表从pandas数据框中select行
可能重复:
如何通过“in”/“in”过滤pandas的dataframe行?
比方说,我有以下pandas数据框:
df = DataFrame({'A' : [5,6,3,4], 'B' : [1,2,3, 5]}) df AB 0 5 1 1 6 2 2 3 3 3 4 5
我可以基于一个特定的值子集:
x = df[df['A'] == 3] x AB 2 3 3
但是,我怎么能基于一个值列表子集? – 像这样的东西:
list_of_values = [3,6] y = df[df['A'] in list_of_values]
这实际上是如何通过“in”/“in”过滤pandas的dataframe行? ,将答案翻译成你的例子给出:
In [5]: df = DataFrame({'A' : [5,6,3,4], 'B' : [1,2,3, 5]}) In [6]: df Out[6]: AB 0 5 1 1 6 2 2 3 3 3 4 5 In [7]: df[df['A'].isin([3, 6])] Out[7]: AB 1 6 2 2 3 3