python - 在数据框中添加新行，并有条件分割前一行

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. reshape stack
3. reset_index用于删除 MultiIndex 的第一级
4. 用新列名称重命名 系列
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