python制作热力图_Python绘制热力图示例

本文实例讲述了Python绘制热力图操作。分享给大家供大家参考，具体如下：

示例一：

# -*- coding: utf-8 -*-

from pyheatmap.heatmap import HeatMap

import numpy as np

N = 10000

X = np.random.rand(N) * 255 # [0, 255]

Y = np.random.rand(N) * 255

data = []

for i in range(N):

tmp = [int(X[i]), int(Y[i]), 1]

data.append(tmp)

heat = HeatMap(data)

heat.clickmap(save_as="1.png") #点击图

heat.heatmap(save_as="2.png") #热图

运行结果：

示例二：

目标：绘制如下的 hotmap 或者 density-map ，把坐标和hot-value或者density-value替换为自己的数据，稍微改进就可以绘制相当不错的热力图（密度图）。

import matplotlib.pyplot as plt

import matplotlib.cm as cm

from matplotlib.colors import LogNorm

import numpy as np

x, y = np.random.rand(10), np.random.rand(10)

z = (np.random.rand(9000000)+np.linspace(0,1, 9000000)).reshape(3000, 3000)

plt.imshow(z+10, extent=(np.amin(x), np.amax(x), np.amin(y), np.amax(y)),

cmap=cm.hot, norm=LogNorm())

plt.colorbar()

plt.show()

希望本文所述对大家Python程序设计有所帮助。