python - 在 Pandas 中创建类似相关矩阵的数据框

Question

我有一个 df，其中包含 A 和 B 的相关值

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

我正在尝试从 DataFrame.corr 给我的类型的 df 创建一个类似相关矩阵的数据框。我试过了

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