熊貓比較列表

Question

我在我的數據框中有列存儲lists，我想比較列中的每個元素與lists。

所有我曾嘗試失敗的方法：

df.list_col == ['3', '4'] 
df.list_col.isin([['3', '4']]) 
df.list_col.equals(['3', '4']) 

有沒有一個簡單的解決方案呢？

Answer 1

您可以使用apply與in：

df = pd.DataFrame({'A':[[1,2],[2,4],[3,1]], 
        'B':[4,5,6]}) 

print (df) 
     A B 
0 [1, 2] 4 
1 [2, 4] 5 
2 [3, 1] 6 

print (df.A.apply(lambda x: 2 in x)) 
0  True 
1  True 
2 False 
Name: A, dtype: bool