如何筛选Pandas DataFrame中列值为指定列表的行？

你遇到的这个错误很典型——Pandas在处理Series（也就是你的col1列）和单个列表直接比较时，会默认尝试按位置逐元素对齐对比，但你的col1是包含4个列表元素的Series，而['a']是长度为1的列表，两者长度不匹配，自然就抛出了ValueError。

下面给你几种可行的解决方案，按需选择：

方法1：通用方案（适用于任意长度的列表） #

使用apply()方法对col1中的每个列表元素单独做判断，这是最通用的方式，不管列表长度是否固定都能用：

import pandas as pd

df = pd.DataFrame({'col1': [['a'], ['b'], ['a'], ['b']]})
# 筛选col1等于['a']的行
filtered_df = df[df['col1'].apply(lambda x: x == ['a'])]
print(filtered_df)

执行后会得到预期结果：

col1
0  [a]
2  [a]

方法2：针对固定长度列表的高效方案（比如你的例子都是单元素列表） #

如果你的col1里每个列表的长度都是固定的（比如都是1个元素），可以用Pandas的序列索引方法，这是向量化操作，比apply()速度更快：

filtered_df = df[df['col1'].str[0] == 'a']

这里str[0]会提取每个列表的第一个元素，然后直接和'a'比较，同样能得到正确的筛选结果。

为什么原来的代码会报错？ #

再补充解释一下：当你执行df['col1'] == ['a']时，Pandas会把右边的['a']当成一个长度为1的序列，尝试和左边长度为4的Series逐位置配对比较——比如拿df['col1'][0]（也就是['a']）和['a'][0]（也就是'a'）对比，df['col1'][1]（['b']）和['a'][1]（但['a']没有第二个元素），这就触发了长度不匹配的错误。

内容的提问来源于stack exchange，提问作者Michael Chao