Python实时变化热图绘制

我有一个2D网格50 * 50.对于每个位置,我有一个强度值(即数据类似于(x,y,强度)这些50 * 50个位置).我想将数据可视化为热图.

扭曲是强度每秒都会改变(对于大多数位置),这意味着我需要每秒重新绘制热图.我想知道什么是处理这种实时变化热图的最佳库/方法.

解决方法:

这实际上取决于您获取数据的方式,但是：

import matplotlib.pyplot as plt
import numpy as np
import time

# create the figure
fig = plt.figure()
ax = fig.add_subplot(111)
im = ax.imshow(np.random.random((50,50)))
plt.show(block=False)

# draw some data in loop
for i in range(10):
    # wait for a second
    time.sleep(1)
    # replace the image contents
    im.set_array(np.random.random((50,50)))
    # redraw the figure
    fig.canvas.draw()

这应该以1秒的间隔绘制11个随机50×50图像.重要的部分是im.set_array替换图像数据和fig.canvas.draw,它将图像重绘到画布上.

如果您的数据实际上是表单中的点列表(x,y,intensity),则可以将它们转换为numpy.array：

import numpy as np

# create an empty array (NaNs will be drawn transparent)
data = np.empty((50,50))
data[:,:] = np.nan

# ptlist is a list of (x, y, intensity) triplets
ptlist = np.array(ptlist)
data[ptlist[:,1].astype('int'), ptlist[:,0].astype('int')] = ptlist[:,2]

标签：matplotlib-basemap,python,matplotlib,ipython
来源： https://codeday.me/bug/20190830/1768978.html