PANDAS :匹配两列并返回索引

Question

我是Python和Pandas的新手。我被一个问题卡住了。我在Excel中有两列，我正在使用PANDAS修改它们

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

我想使用PANDAS返回所有值Yes，所以这里我想要一个包含所有yes的列表，即ABC、OAI、PAU

请大家帮帮我，我是Python新手。

Answer 1

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']