python - 将数据框关联到 pandas 中的图形

Question

dataframe我用这段代码计算相关性：

corr = df.corr()
corr.style.background_gradient(cmap='coolwarm')

我得到了这个结果（截图）：

然而，这导致表格形式而不是数字。如何将我的相关结果转换为 JPG 或 PNG？

标签： pythonpandas

Answer 1

你可以试试这个

import matplotlib.pyplot as plt
import seaborn as sns

fig, ax = plt.subplots()
sns.heatmap(df.corr(), annot=True, fmt='.4f', 
            cmap=plt.get_cmap('coolwarm'), cbar=False, ax=ax)
ax.set_yticklabels(ax.get_yticklabels(), rotation='horizontal')
plt.savefig(YOUR_PATH, bbox_inches='tight', pad_inches=0.0)