借鉴原文如下:https://blog.csdn.net/coder_Gray/article/details/81867639
真的是找了两天才找到既简洁又能满足我全部需求的例子,感谢作者了。
根据需求做了如下修改
from matplotlib import pyplot as plt
from matplotlib import cm
from matplotlib import axes
plt.rcParams['font.sans-serif'] = ['SimHei'] # 用来正常显示中文
plt.rcParams['axes.unicode_minus'] = False # 用来正常显示负号
def draw(xLabel,yLabel,heatlist,title):
#作图阶段
fig = plt.figure(dpi =200)
#定义画布为1*1个划分,并在第1个位置上进行作图
ax = fig.add_subplot(111)
#定义横纵坐标的刻度
ax.set_xticks(np.arange(len(xLabel)), labels=xLabel)
ax.set_yticks(np.arange(len(yLabel)), labels=yLabel)
#作图并选择热图的颜色填充风格,这里选择hot
im = ax.imshow(heatlist, cmap=plt.cm.hot_r)
# Rotate the tick labels and set their alignment.
plt.setp(ax.get_xticklabels(), rotation=45, ha="right",rotation_mode="anchor")
# Loop