Python数据操作：从一列中拆分数据以在同一数据框中创建更多行

Question

所以我的输入是：

   Col1         Col2      Col3 Col4
0   123  abc,def,ghi  Country1  XXX
1   456      pqr,stu  Country2   XX
2   789          xyz  Country2   YY

我希望我的输出为：

   Col1      Col2    Col3   Col4
0    abc     123  Country1    XXX
1    def     123  Country1    XXX
2    ghi     123  Country1    XXX
3    pqr     456  Country2     XX
4    stu     456  Country2     XX
5    xyz     789  Country2     YY

请问最pythonic的方法是什么？谢谢vm！

Answer 1

您可以使用str.split 和stack 为join 创建Series 到原始DataFrame：

print (df.Col2
      .str
      .split(',',expand=True)
      .stack()
      .reset_index(drop=True, level=1)
      .rename('Col2'))

0    abc
0    def
0    ghi
1    pqr
1    stu
2    xyz
Name: Col2, dtype: object


print (df.drop('Col2', axis=1)
             .join
             (
             df.Col2
             .str
             .split(',',expand=True)
             .stack()
             .reset_index(drop=True, level=1)
             .rename('Col2')           
             ))

   Col1      Col3 Col4 Col2
0   123  Country1  XXX  abc
0   123  Country1  XXX  def
0   123  Country1  XXX  ghi
1   456  Country2   XX  pqr
1   456  Country2   XX  stu
2   789  Country2   YY  xyz

Answer 2

使用extractall 和join

d1 = df.Col2.str.extractall('([^,]+)') \
       .rename(columns={0: 'Col2'}) \
       .reset_index(1, drop=True)

df.drop('Col2', 1).join(d1).reindex_axis(df.columns, 1)