使用pandas创建类似相关矩阵的数据框

Question

我有一个包含A和B相关性值的数据框（df）

df = pd.DataFrame({'x':['A','A','B','B'],'y':['A','B','A','B'],'c':[1,0.5,0.5,1]})

我尝试从df创建一个类似于DataFrame.corr的相关矩阵数据框架。

我已经尝试过：

corr = df.pivot_table(columns='y',index='x')
y    A    B
x          
A  1.0  0.5
B  0.5  1.0

但我不知道如何去除多级索引。

- HappyPy

Answer 1

您只需要指定values即可摆脱多重索引。

corr = df.pivot_table(columns='y',index='x', values='c')

Out[41]:
y    A    B
x
A  1.0  0.5
B  0.5  1.0

如果你也想摆脱轴名称，可以使用rename_axis方法

corr = (df.pivot_table(columns='y',index='x', values='c')
          .rename_axis(index=None, columns=None))

Out[43]:
     A    B
A  1.0  0.5
B  0.5  1.0

- Andy L.