在数据框中添加新行并有条件拆分前一行

Question

如果特定列值中存在“或”条件，则它必须通过数据框中的先前值拆分并创建新行。尝试拆分和连接无法获得所需的输出，非常感谢任何帮助
输入

  col1   col2           col3              col4
0  x     bca or cba     value1 or null    x1 or x2
1  y     bca            value2            x1

输出

   col1   col2           col3        col4
 0  x     bca            value1      x1
 1  x     cba            null        x2
 2  y     bca            value2      x1

Answer 1

用途：

cols = df.columns

df = (df.join(df.pop('col2')
        .str.split(' or ', expand=True)
        .stack()
        .reset_index(level=1, drop=True)
        .rename('col2'))
      ).reset_index(drop=True).reindex(columns=cols)
print (df)
  col1 col2    col3
0    x  bca  value1
1    x  cba  value1
2    y  bca  value2

解释：

1. 第一个pop 列用于提取split 列与expand = True 到DataFrame
2. 由stack重塑
3. reset_index 用于删除 MultiIndex 的第一级
4. rename Series 按新列名
5. 然后join 到原来的df
6. 函数 reset_index 是唯一索引所必需的
7. 最后一个reindex 用于相同顺序的列名称

编辑：

df = (df.set_index('col1')
        .stack()
        .str.split(' or ', expand=True)
        .stack()
        .unstack(1)
        .reset_index(level=1, drop=True)
        .reset_index()

)
print (df)

  col1 col2    col3 col4
0    x  bca  value1   x1
1    x  cba    null   x2
2    y  bca  value2   x1