首页 > 编程笔记 > Python笔记

Python imshow()函数:绘制热力图

热力图是一种用于展示密度函数的图。

Python 中,可以通过 matplotlib.pyplot 模块中的 imshow() 函数绘制热力图,其语法格式如下:

imshow(X,cmap,aspect,alpha,origin)


示例代码如下:
import matplotlib.pyplot as plt
import numpy as np
#显示中文
plt.rcParams['font.sans-serif']='SimHei'
#显示负号
plt.rcParams['axes.unicode_minus']=False
#x轴的刻度线向内显示
plt.rcParams['xtick.direction']='in'
#y轴的刻度线向外显示
plt.rcParams['ytick.direction']='out'
#创建画布
plt.figure(figsize=(12,10))
#热力图标题
plt.title('线上平台销售量分析表')
#数据
x=['淘宝','当当','天猫','京东']
y=['C语言教程','Java教程','Python教程','Go语言教程']
data=np.array([[84511,75884,56887,95774],[75488,87454,66874,58777],[98544,58744,
68544,42544],[101147,95220,78554,68744]])
#绘制热力图
plt.imshow(data)
plt.xticks(np.arange(len(x)),labels=x)
plt.yticks(np.arange(len(y)),labels=y)
#设置颜色条
plt.colorbar()
plt.show()
上面代码的运行结果如下图所示。

推荐阅读