python - 在 Pandas 中转换数据框

Question

假设我有数据框:

x1  x2  value_x1    value_x2
a   b   1           2       
a   c   3           4
b   c   5           6

我能得到下面的数据框吗？

a   b   c
1   2   0
3   0   4
0   5   6

我尝试了 pivot_table 和 melt，但结果不合适。

谢谢。

Answer 1

使用wide_to_long :

df = pd.wide_to_long(df.reset_index(), stubnames=['x', 'value_x'], i='index', j='a')

print (df)
         x  value_x
index a            
0     1  a        1
1     1  a        3
2     1  b        5
0     2  b        2
1     2  c        4
2     2  c        6

然后 unstack进行一些数据清理:

df = (df.reset_index(level=1, drop=True)
        .set_index('x', append=True)['value_x']
        .unstack(fill_value=0)
        .rename_axis(None)
        .rename_axis(None, axis=1))
print (df)
   a  b  c
0  1  2  0
1  3  0  4
2  0  5  6