如何使用Python繪制熱力圖

這篇文章將為大家詳細(xì)講解有關(guān)如何使用Python繪制熱力圖，文章內(nèi)容質(zhì)量較高，因此小編分享給大家做個(gè)參考，希望大家閱讀完這篇文章后對(duì)相關(guān)知識(shí)有一定的了解。

示例一：

# -*- 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") #點(diǎn)擊圖
heat.heatmap(save_as="2.png") #熱圖

運(yùn)行結(jié)果：

示例二：

目標(biāo)：繪制如下的 hotmap 或者 density-map ，把坐標(biāo)和hot-value或者density-value替換為自己的數(shù)據(jù)，稍微改進(jìn)就可以繪制相當(dāng)不錯(cuò)的熱力圖（密度圖）。

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()

關(guān)于如何使用Python繪制熱力圖就分享到這里了，希望以上內(nèi)容可以對(duì)大家有一定的幫助，可以學(xué)到更多知識(shí)。如果覺得文章不錯(cuò)，可以把它分享出去讓更多的人看到。