在大熊猫中创建类似相关矩阵的数据框

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

但是我不知道如何摆脱多重索引。

Answer 1

您只需要指定values即可摆脱multiindex

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