python - Pandas 'partial melt' 或 'group melt'

Question

我有一个这样的DataFrame

>>> df = pd.DataFrame([[1,1,2,3,4,5,6],[2,7,8,9,10,11,12]], 
                      columns=['id', 'ax','ay','az','bx','by','bz'])
>>> df
   id  ax  ay  az  bx  by  bz
0   1   1   2   3   4   5   6
1   2   7   8   9  10  11  12

我想把它改造成这样的东西

   id name   x   y   z
0   1    a   1   2   3
1   2    a   7   8   9
2   1    b   4   5   6
3   2    b  10  11  12

这是一个逆轴/融化问题，但我不知道有什么方法可以通过保持这些组的完整性来融化。我知道我可以在原始数据帧上创建投影，然后 concat 那些，但我想知道我是否遗漏了我的工具带中的一些常见融化技巧。

Answer 1

set_index，将列转换为多索引并堆叠，

df = df.set_index('id')
df.columns = [df.columns.str[1], df.columns.str[0]]
new_df = df.stack().reset_index().rename(columns = {'level_1': 'name'})

    id  name    x   y   z
0   1   a       1   2   3
1   1   b       4   5   6
2   2   a       7   8   9
3   2   b       10  11  12