python中heatmap 函数可视化矩阵热图的变量名改为中文
时间: 2023-11-12 17:03:45 浏览: 146
Heatmap_Dendrogram_Python:国家/地区数据的热图和树状图
要将热图的变量名改为中文,你可以使用 Matplotlib 的字体设置功能来实现。下面是一个示例代码:
```python
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
from matplotlib.font_manager import FontProperties
# 创建一个包含多个变量的数据框
data = pd.DataFrame({
'变量1': [1, 2, 3, 4, 5],
'变量2': [2, 4, 6, 8, 10],
'变量3': [3, 6, 9, 12, 15]
})
# 计算相关系数矩阵
correlation_matrix = data.corr()
# 设置中文字体
font = FontProperties(fname='SimHei.ttf', size=12) # 替换为你的中文字体文件路径和字号
# 使用热图可视化相关系数矩阵
sns.heatmap(correlation_matrix, annot=True, cmap='coolwarm')
# 修改 x 轴和 y 轴的标签为中文
plt.xticks(range(len(correlation_matrix.columns)), correlation_matrix.columns, fontproperties=font)
plt.yticks(range(len(correlation_matrix.columns)), correlation_matrix.columns, fontproperties=font)
# 显示图表
plt.show()
```
在上面的代码中,我们通过 `FontProperties` 类来设置中文字体,并将其传递给 `xticks` 和 `yticks` 函数的 `fontproperties` 参数,以替换 x 轴和 y 轴的标签为中文。你需要将 `fname` 参数替换为你的中文字体文件的路径,同时可以调整 `size` 参数来控制字号。
请确保在你的代码所在目录中存在正确的中文字体文件,并将 `'SimHei.ttf'` 替换为你实际使用的中文字体文件名。
这样,你就可以将热图的变量名改为中文来进行可视化了。
阅读全文