如何使用过滤器提取列

Question

数据框在下面

import pandas as pd
import io
s = '''uid,col1,flag
1001,rejected,Audi
1002,verified,Benz
1003,verified,Ferrari
1004,rejected,Tesla'''
df_j = pd.read_csv(io.StringIO(s))

需要提取verified所在的行 预期结果低于

1002
1003

Answer 1

您可以结合使用boolean indexing和DataFrame.loc进行选择：

s = df_j.loc[df_j['col1'] == 'verified', 'uid']
print (s)
1    1002
2    1003
Name: uid, dtype: int64