如果您打算为情节添加颜色和标签,您可以试试这个:
import seaborn as sns
ax = sns.heatmap(corr,vmin=-1, vmax=1, center=0,cmap=sns.diverging_palette(20, 220, n=200),square=True)
ax.set_xticklabels(ax.get_xticklabels(),rotation=45,horizontalalignment='right')
这里的 cmap 参数处理从数据值到颜色空间的映射。