Python：将两个链接的列拆分为新行

Question

我有以下数据框：

N1   FF1; FF2                PP1; PP2
N2   FF3                     PP3
N3   FF4; FF5; FF6           PP4; PP5; PP6

列FF和PP有以下关系，每个FF都链接到具有相同编号的PP，即FF1链接到PP1等等。

我需要的输出如下：

N1 FF1 PP1
N1 FF2 PP2
N2 FF3 PP3
N3 FF4 PP4
N3 FF5 PP5
N3 FF6 PP6

到目前为止，我已经尝试了以下方法：

df_copy = df["F","P"].str.split(";").apply(Series,1).stack()

但是我收到一个 KeyError 并且对如何继续感到困惑...

对编程还很陌生，非常感谢一些指导。

谢谢！

Answer 1

对于 Pandas / NumPy，您应该尽可能使用矢量化方法。 apply + pd.Series 是一个低效的 Python 级循环。

例如，使用itertools.chain and np.repeat:

from itertools import chain

df = pd.DataFrame([['N1', 'FF1; FF2', 'PP1; PP2'],
                   ['N2', 'FF3', 'PP3'],
                   ['N3', 'FF4; FF5; FF6', 'PP4; PP5; PP6']])

split1, split2 = df[1].str.split('; '), df[2].str.split('; ')
n = split1.str.len()

res = pd.DataFrame({0: df[0].values.repeat(n.values),
                    1: list(chain.from_iterable(split1)),
                    2: list(chain.from_iterable(split2))})

print(res)

    0    1    2
0  N1  FF1  PP1
1  N1  FF2  PP2
2  N2  FF3  PP3
3  N3  FF4  PP4
4  N3  FF5  PP5
5  N3  FF6  PP6

Python：将两个链接的列拆分为新行

Python: Splitting two linked columns into new rows

python

data-manipulation

python-2.7

pandas

keyerror