python – Pandas将数据框展平为单个列

我有以下格式的数据集：

df = pd.DataFrame({'x':[1,2,3], 'y':[10,20,30], 'v1':[3,2,3] , 'v2':[13,25,31] })

>> v1 v2  x   y
   3  13  1  10
   2  25  2  20
   3  31  3  30

使用x设置索引列,我希望将v1和v2(V)组合的数据展平,预期输出如下：

>> x   y   V
   1  10   3
   1  10   13
   2  20   2
   2  20   25
   3  30   3
   3  30   31

再次使用df的原始格式.我尝试使用堆栈和取消堆栈进行重新整形,但我无法按照我期望的方式进行.

非常感谢！

解决方法:

您可以将stack与set_index一起使用.最后drop列level_2：

print (df.set_index(['x','y']).stack().reset_index(name='V').drop('level_2', axis=1))
   x   y   V
0  1  10   3
1  1  10  13
2  2  20   2
3  2  20  25
4  3  30   3
5  3  30  31

melt和sort_values的另一个解决方案：

print (pd.melt(df, id_vars=['x','y'], value_name='V')
         .drop('variable', axis=1)
         .sort_values('x'))

   x   y   V
0  1  10   3
3  1  10  13
1  2  20   2
4  2  20  25
2  3  30   3
5  3  30  31