查找多个模式并合并

Question

如何在新的数据框中查找和排序以下内容

输入：

      IdA        IdB           Value
0     123        ABC_25_01     1
1     456        DEF_25_01     2
2     123        ABC_60_01     3
3     456        DEF_60_01     4

输出

      IdA        IdB      Value25 Value60
0     123        ABC_01   1       3
1     456        DEF_01   2       4

谢谢！

Answer 1

假设IdB 中的值都采用X_Y_Z 的形式，其中X_Z 是最终的IdB，Y 是列标题：

1. 使用str.extract 获取“类别”，即两个下划线之间的数字
2. 用户 str.replace 从 IdB 中删除“类别”。
3. pivot 获取所需的输出

df["Category"] = df["IdB"].str.extract("_(.*)_")
df["IdB"] = df["IdB"].str.replace("_(.*)_", "_", regex=True)
output = (df.pivot(["IdA", "IdB"], "Category", "Value")
          .add_prefix("Value")
          .reset_index())

>>> output

Category  IdA     IdB  Value25  Value60
0         123  ABC_01        1        3
1         456  DEF_01        2        4