如果列是列表，是否可以過濾Pandas DataFrame列？

Question

import pandas as pd 

dafr = pd.DataFrame({'a': [1,2,3], 'b': [[1,2,3],[2,3,4],[3,4,5]]}) 

我試着這樣做：

dafr[dafr['b'].isin(2)] 

應返回行有列出：[1,2,3] & [2,3,4]。

我想知道這是否可能？

Answer 1

isin返回列值是否在您傳遞的值中。你想檢查你傳遞的值是否在列值中。

據我所知，這個沒有直接的快捷方式，但可以使用map做到這一點：

>>> dafr[dafr.b.map(lambda x: 2 in x)] 
    a   b 
0 1 [1, 2, 3] 
1 2 [2, 3, 4] 

Answer 2

dafr[dafr['b'].apply(lambda x: 2 in x)] 

Answer 3

如果存儲在B元組的列，而不是表的列，afr[dafr['b'].apply(lambda x: 2 in x)] 將執行得相當快。