python - 根据 Pandas 列中的列表值从 DataFrame 中选择行

Question

我是 pandas 的新手，我有一个简单的数据框，想根据列提取某些行。但是，此列中的类型是列表。

例子:

df = pd.DataFrame([['text1', [1,2,3]], ['text2', [2,3,4]]], columns=['text','list_value'])

数据框看起来像:

    text    list_value
0   text1   [1, 2, 3]
1   text2   [2, 3, 4]

我试过了

df.loc[df['list_value'] == [1,2,3]]

它返回一个错误:

ValueError: Arrays were different lengths: 2 vs 3

我想知道是否有比使用 for 循环迭代数据帧更好的解决方案。

类似的问题，但解决方案对我不起作用:Select rows from a DataFrame based on values in a column in pandas .

Answer 1

您可以添加apply tuple，当单元格中有列表时，pandas 有时会返回有线结果

df.loc[df['list_value'].apply(tuple) == tuple([1,2,3])]
Out[58]: 
    text list_value
0  text1  [1, 2, 3]