使用data.explode(['A','B'])报错ValueError：列元素数量不匹配的原因？

先看创建目标DataFrame的代码：

import pandas as pd

list1 = [1, 6, 7, [46, 56, 49], 45, [15, 10, 12]]
list2 = [[49, 57, 45], 3, 7, 8, [16, 19, 12], 41]

data = {'A': list1, 'B': list2}
data = pd.DataFrame(data)

执行连续两次单列explode可以正常运行：

data.explode('A').explode('B')

但直接传入多列列表执行explode会触发错误：

data.explode(['A', 'B'])

错误信息如下：

ValueError                                Traceback (most recent call last)
<ipython-input-97-efafc6c7cbfa> in <module>
      5         'B': list2}
      6 data = pd.DataFrame(data)
----> 7 data.explode(['A', 'B'])

~\AppData\Roaming\Python\Python38\site-packages\pandas\core\frame.py in explode(self, column, ignore_index)
   9033             for c in columns[1:]:
   9034                 if not all(counts0 == self[c].apply(mylen)):
-> 9035                     raise ValueError("columns must have matching element counts")
   9036             result = DataFrame({c: df[c].explode() for c in columns})
   9037         result = df.drop(columns, axis=1).join(result)

ValueError: columns must have matching element counts

错误原因解析 #

两者的核心差异在于pandas对多列explode的规则限制：

• 当同时对多列执行explode时，pandas要求每一行中，所有待爆炸列的可迭代元素长度必须完全一致（单个值会被视为长度1的元素）。
• 看你的DataFrame各行的元素长度匹配情况：
  • 第0行：A是单个值（长度1），B是长度3的列表 → 长度不匹配
  • 第3行：A是长度3的列表，B是单个值（长度1） → 长度不匹配
  • 第4行：A是单个值（长度1），B是长度3的列表 → 长度不匹配
  • 第5行：A是长度3的列表，B是单个值（长度1） → 长度不匹配
    这些行的列长度不一致，直接触发了ValueError。
• 而连续两次单列explode是分步处理：先爆炸A列，把A的列表拆成多行，对应行的B值会被重复对应次数；接着再爆炸B列，处理此时的B列元素，完全不限制原行的长度匹配，因此可以正常运行。

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