python - Pandas 从列中可用的列表数据中扩展行

Question

我在 pandas 中有一个这样的数据框:

 column1      column2
 [a,b,c]        1
 [d,e,f]        2
 [g,h,i]        3

预期输出:

column1      column2
  a              1
  b              1
  c              1
  d              2
  e              2
  f              2
  g              3
  h              3
  i              3

如何处理这些数据？

Answer 1

DataFrame.explode

自 pandas >= 0.25.0我们有 explode为此的方法，它将列表扩展为每个元素的一行并重复其余列:

df.explode('column1').reset_index(drop=True)

输出

  column1  column2
0       a        1
1       b        1
2       c        1
3       d        2
4       e        2
5       f        2
6       g        3
7       h        3
8       i        3

---

自 pandas >= 1.1.0我们有 ignore_index 参数，所以我们不必与 reset_index 链接:

df.explode('column1', ignore_index=True)

输出

  column1  column2
0       a        1
1       b        1
2       c        1
3       d        2
4       e        2
5       f        2
6       g        3
7       h        3
8       i        3