python – 使用pandas绘制相关矩阵

我有一个具有大量功能的数据集,因此分析相关矩阵变得非常困难.我想绘制一个相关矩阵,我们使用pandas库中的dataframe.corr()函数.是否有任何内置函数由pandas库提供以绘制此矩阵？

解决方法:

您可以使用matplotlib中的pyplot.matshow()：

import matplotlib.pyplot as plt

plt.matshow(dataframe.corr())
plt.show()

编辑：

在评论中提出了如何更改轴刻度标签的请求.这是一个豪华版本,绘制在更大的图形尺寸上,具有与数据框匹配的轴标签,以及用于解释颜色比例的颜色条图例.

我包括如何调整标签的大小和旋转,我使用的是图形比率,使色条和主要图形的高度相同.

f = plt.figure(figsize=(19, 15))
plt.matshow(df.corr(), fignum=f.number)
plt.xticks(range(df.shape[1]), df.columns, fontsize=14, rotation=45)
plt.yticks(range(df.shape[1]), df.columns, fontsize=14)
cb = plt.colorbar()
cb.ax.tick_params(labelsize=14)
plt.title('Correlation Matrix', fontsize=16);