查找存储在数据框单元格中的列表

Question

我有一个像这样的数据框df：

    values
0    [0,1]
1    [1,2]
2    [2,3]
3    [3,4]
4    [4,5]

类似的事情不起作用：

df[df['values'] == [0,1]]

我得到：

ValueError: 'Lengths must match to compare', (5,), (2,)

给定存储在该行中的list，如何获得一行？

Answer 1

使用列表理解：

[i == [0,1] for i in df['values']]

赞：

df[[i == [0,1] for i in df['values']]]

输出：

   values
0  [0, 1]

Answer 2

df.loc[(df['values'].apply(pd.Series) == [0,1]).all(axis=1)]