使用值列表从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