热力图,即 heatmap,是一种常用的图表,如下图所示。

热力图通过颜色的深浅反映数据的大小,横坐标代表一个指标,纵坐标代表一个指标。
用 Python 绘制热力图非常简单,只需要调用 seaborn 中的 heatmap 方法即可。
首先附上代码,关键代码的含义请参考注释。
import matplotlib.pyplot as plt
import seaborn as sns
import numpy as np
import pandas as pd
%matplotlib notebook
from pylab import mpl
# 设置字体,防止中文乱码
mpl.rcParams['font.sans-serif'] = ['FangSong']
# 生成一个3x3的随机矩阵
arr = np.random.rand(3, 3)
fig, ax = plt.subplots(figsize = (6, 5))
# 调动heatmap方法,annot表示注释,即方格中的数字,cmap表示颜色代码
sns.heatmap(pd.DataFrame(np.round(arr, 2)), annot=True, cmap="YlGnBu")
# 设置标题、坐标轴标签及字体大小
ax.set_title('热力图', fontsize = 14)
ax.set_ylabel('Y', fontsize = 14)
ax.set_xlabel('X', fontsize = 14)
plt.show()
运行结果:

可以看到,用 Python 绘制热力图非常简单。
觉得有用,点赞+收藏哦!