PANDAS:匹配两列并返回索引

时间:2020-04-28 18:46:42

标签: python pandas

我是Python和Pandas的新手。我陷入一个问题。 我在Excel中有两列正在使用PANDAS进行修改

0 ABC Yes
1 TZY No
2 KYZ No
3 HJO No
4 OAI Yes
5 PAU yes

我想使用PANDAS返回所有值为Yes 所以在这里,我要列出所有是的列表,即[ABC,OAI,PAU]

任何人都可以帮助我,我是Python的新手。

1 个答案:

答案 0 :(得分:2)

df = pd.read_clipboard(header=None)

print(df)

    0   1   2
0   0   ABC Yes
1   1   TZY No
2   2   KYZ No
3   3   HJO No
4   4   OAI Yes
5   5   PAU yes

yes_list = ['Yes', 'yes']
df = df[df[2].isin(yes_list)]

print(df)


0   1   2
0   0   ABC Yes
4   4   OAI Yes
5   5   PAU yes

如果您只想使用列表而不是df分片,则这里有一些用于编辑的其他代码。

yes_value_list = df[1].to_list()

print(yes_value_list)
['ABC', 'OAI', 'PAU']
相关问题