基于列名前缀的 Pandas 数据透视列

Question

我有一个数据框:

df = AG_Speed AG_wolt AB_Speed AB_wolt C1 C2 C3
       1         2      3         4     6  7  8
       1         9      2         6     4  1  8

我想根据前缀对它进行透视以获得:

df = Speed Wolt C1 C2 C3 Category
      1      2   6 7  8    AG
      3      4   6 7  8    AB
      1      9   4 1  8    AG
      2      6   4 1  8    AG 

最好的方法是什么？

Answer 1

我们可以使用pd.wide_to_long为了这。但是因为它希望列名以 stub 名称开头​​，所以我们必须反转列格式:

df.columns = ["_".join(col.split("_")[::-1]) for col in df.columns]
res = pd.wide_to_long(
    df, 
    stubnames=["Speed", "wolt"], 
    i=["C1", "C2", "C3"], 
    j="Category", 
    sep="_", 
    suffix="[A-Za-z]+"
).reset_index()

   C1  C2  C3 Category  Speed  wolt
0   6   7   8       AG      1     2
1   6   7   8       AB      3     4
2   4   1   8       AG      1     9
3   4   1   8       AB      2     6

如果您希望列按特定顺序排列，请使用 DataFrame.reindex:

res.reindex(columns=["Speed", "wolt", "C1", "C2", "C3", "Category"])

   Speed  wolt  C1  C2  C3 Category
0      1     2   6   7   8       AG
1      3     4   6   7   8       AB
2      1     9   4   1   8       AG
3      2     6   4   1   8       AB