熊猫爆炸多列[重复]

Question

我的 DF 有多个列。 其中两列是相同 len 的列表。（col2 和 col3 是列表。列表的 len 相同）。

我的目标是在自己的行上列出每个元素。

我可以使用df.explode()。但它只接受一列。但是，我希望这两列对“爆炸”。如果我先执行df.explode('col2')，然后执行df.explode('col3')，则结果为 9 行而不是 3 行。

原始DF

col0      col1        col2        col3
1       aa          [1,2,3]     [1.1,2.2,3.3]
2       bb          [4,5,6]     [4.4,5.5,6.6]
3       cc          [7,8,9]     [7.7,8.8,9.9]
3       cc          [7,8,9]     [7.7,8.8,9.9]

结束数据帧

id      col1        col2        col3
1       aa          1           1.1
1       aa          2           2.2
1       aa          3           3.3
2       bb          4           4.4
2       bb          5           5.5
2       bb          6           6.6
3       cc          ...         ...

更新 所有列都没有唯一值，因此不能用作索引。

Answer 1

您可以将col1 设置为索引并将pd.Series.explode 应用于列：

df.set_index('col1').apply(pd.Series.explode).reset_index()

或者：

df.apply(pd.Series.explode)


   col1 col2 col3
0    aa    1  1.1
1    aa    2  2.2
2    aa    3  3.3
3    bb    4  4.4
4    bb    5  5.5
5    bb    6  6.6
6    cc    7  7.7
7    cc    8  8.8
8    cc    9  9.9
9    cc    7  7.7
10   cc    8  8.8
11   cc    9  9.9

Answer 2

我从其他答案中借用了这个解决方案（忘记在哪里）：

df.explode(['col2', 'col3']).

优势：比apply 解决方案更快。

确保 col2 和 col3 在同一行的单元格中具有相同数量的元素。