使用大熊猫绘制最高的相关性

Question

我一直在使用以下代码通过热图绘制相关性。但是，变量太多。是否可以在图表上绘制最高相关性（超过0.5和-.5）？

plt.rcParams['figure.figsize'] = [80,80]
corr3 = datasetcm.corr()


fig = plt.figure()
ax = fig.add_subplot(111)
cax = ax.matshow(corr3,cmap='coolwarm', vmin=-1, vmax=1)
fig.colorbar(cax)
ticks = np.arange(0,len(datasetcm.columns),1)
ax.set_xticks(ticks)
plt.xticks(rotation=90)
ax.set_yticks(ticks)
ax.set_xticklabels(datasetcm.columns)
ax.set_yticklabels(datasetcm.columns)
plt.show()

Answer 1

在绘制之前将相关矩阵过滤为0.5的阈值。对于低于0的相关性，这将返回0.5。

然后，我们可以使用颜色映射将0的行显示为not correlated

corr3 = datasetcm.corr()
corr3 = corr3 [corr3 > 0.5].fillna(0)

corr3.style.background_gradient(cmap='coolwarm', axis=None).set_precision(2)