如何在Pandas数据框中匹配列表值？解决长度不匹配报错

问题原因 #

执行df.loc[df['B']== [1,3,2]]报错，是因为Pandas会把右侧的列表当成长度为3的序列，试图和长度为4的B列逐元素对齐比较，因长度不匹配抛出"Lengths must match to compare"错误。而匹配标量（比如4）时，Pandas会自动把标量广播到整个序列，所以能正常运行。

解决方案 #

以下两种方法可以正确匹配序列中的列表：

方法1：用apply遍历元素逐个比较

通过apply对B列每个元素做列表相等性判断：

result = df.loc[df['B'].apply(lambda x: x == [1,3,2])]
print(result)

输出结果：

A         B
0  2  [1, 3, 2]

方法2：先判断类型再比较（更严谨）

如果B列混合了列表和标量，可先筛选出列表类型的元素再做比较，避免无效的类型对比：

result = df.loc[df['B'].apply(lambda x: isinstance(x, list) and x == [1,3,2])]
print(result)

该方法仅对列表类型元素做相等性检查，标量元素直接返回False，最终结果和方法1一致。

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