使用值列表从 Pandas 数据框中选择行 [英] Use a list of values to select rows from a Pandas dataframe
本文介绍了使用值列表从 Pandas 数据框中选择行的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
假设我有以下 Pandas 数据框:
Let’s say I have the following Pandas dataframe:
df = DataFrame({'A' : [5,6,3,4], 'B' : [1,2,3, 5]})
df
A B
0 5 1
1 6 2
2 3 3
3 4 5
我可以基于特定值进行子集化:
I can subset based on a specific value:
x = df[df['A'] == 3]
x
A B
2 3 3
但是如何根据值列表进行子集化?- 像这样:
But how can I subset based on a list of values? - something like this:
list_of_values = [3,6]
y = df[df['A'] in list_of_values]
获得:
A B
1 6 2
2 3 3
推荐答案
您可以使用 isin
方法:
You can use the isin
method:
In [1]: df = pd.DataFrame({'A': [5,6,3,4], 'B': [1,2,3,5]})
In [2]: df
Out[2]:
A B
0 5 1
1 6 2
2 3 3
3 4 5
In [3]: df[df['A'].isin([3, 6])]
Out[3]:
A B
1 6 2
2 3 3
相反,使用~
:
In [4]: df[~df['A'].isin([3, 6])]
Out[4]:
A B
0 5 1
3 4 5
这篇关于使用值列表从 Pandas 数据框中选择行的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!
查看全文