如果column是列表，是否可以过滤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)]将执行得非常快。