如何使用pandas过滤器和unique测试if循环？

Question

我有一个大的数据框，我想检查一列的过滤器在不同的列中是否存在唯一的值

import pandas

df = pandas.DataFrame({'Site Name' : ['NYC','NYC','NYC','ATL'], 'Product Family' : [5,2,3,2]})
name = 'NYC'

if 5 in df[(df['Site Name'] == name & df['Product Family'].unique)]:
    print('All good')
else:
    print('nope')

我收到错误类型错误：&；的操作数类型不受支持：'str和method

我想过滤df['Site Name']中的一个值（'NYC'），然后检查所有行中不同列df['Product Family']中是否存在值（5）

Answer 1

您可能需要添加any

if any(df['Site Name'].isin([name])&df['Product Family'].isin([5])):
    print('All good')
else:
    print('nope')

All good