python - 如果它们满足 Pandas 中的条件，如何将列的唯一值存储为列表？

Question

如果它们满足条件，我正在尝试将列的唯一值保存为列表。我的语法/逻辑错误。我如何以正确的方式获得它？

输入：

ID  Proceed?
ABC  Yes
DEF  No
EFG  Yes
HHH  No
JJJ  Yes
NNN  No

预期输出：

['ABC','EFG','JJJ']

代码：

a = list(df1['ID'].unique().where df1['Proceed?'] == 'Yes')

标签： pythonpandas

Answer 1

我会去

a = df1[df1['Proceed?'] == 'Yes']['ID'].unique().tolist()