Python - Pandas DataFrame中的Unnest单元格

Question

假设我有DataFrame df:

a b c
v f 3|4|5
v 2 6
v f 4|5

我想生产这个df:

a b c
v f 3
v f 4
v f 5
v 2 6
v f 4
v f 5

我知道如何使用tidyr包在R中进行这种转换.

在熊猫中有这么简单的方法吗？

Answer 1

你可以：

import numpy as np

df = df.set_index(['a', 'b'])
df = df.astype(str) + '| ' # There's a space ' ' to match the replace later
df = df.c.str.split('|', expand=True).stack().reset_index(-1, drop=True).replace(' ', np.nan).dropna().reset_index() # and replace also has a space ' '

要得到：

   a  b  0
0  v  f  3
1  v  f  4
2  v  f  5
3  v  2  6
4  v  f  4
5  v  f  5