如何将数据帧唯一值与列表进行比较？

Question

我有一个Panda数据框列，我想检查列中的所有值是否都来自另一个列表。

例如，我想检查我的列中的所有值是否为A或B而没有其他值。我的代码应为以下输入返回true：

myValues = ['A','B'] 
df = pd.DataFrame(['A','B','B','A'],columns=['Col']) # True 
df = pd.DataFrame(['A','A'],columns=['Col']) # True 
df = pd.DataFrame(['B'],columns=['Col']) # True 
df = pd.DataFrame(['B','C'],columns=['Col']) # False 

Answer 1

您可以使用isin并通过您的列表来生成一个布尔值数组，并与all返回所有存在的值是否：

In [146]:  
myValues = ['A','B'] 
df = pd.DataFrame(['A','B','B','A'],columns=['Col']) # True 
print(df['Col'].isin(myValues).all()) 
df = pd.DataFrame(['A','A'],columns=['Col']) # True 
print(df['Col'].isin(myValues).all()) 
df = pd.DataFrame(['B'],columns=['Col']) # True 
print(df['Col'].isin(myValues).all()) 
df = pd.DataFrame(['B','C'],columns=['Col']) # False 
print(df['Col'].isin(myValues).all()) 


True 
True 
True 
False 

Answer 2

这里是一个替代的解决方案：

df.eval('Col in @myValues') 

演示：

In [78]: pd.DataFrame(['A','B','B','A'],columns=['Col']).eval('Col in @myValues') 
Out[78]: 
0 True 
1 True 
2 True 
3 True 
dtype: bool 

In [79]: pd.DataFrame(['A','A'],columns=['Col']).eval('Col in @myValues') 
Out[79]: 
0 True 
1 True 
dtype: bool 

In [80]: pd.DataFrame(['B'],columns=['Col']).eval('Col in @myValues') 
Out[80]: 
0 True 
dtype: bool 

In [81]: pd.DataFrame(['B','C'],columns=['Col']).eval('Col in @myValues') 
Out[81]: 
0  True 
1 False 
dtype: bool