如何在随机游走中使用 imshow 和 colorbar 函数？

编程问答 2022-05-03

问题描述

这是我的随机游走生成器函数代码：

import random
def random_walk():
    a = b = 0
    lst = [ (0,1),(0,-1),(1,0),(-1,0) ]
    while True:
        yield (a,b)
        da,db = random.choice(lst)
        a,b   = a+da,b+db
        
from itertools import islice
print(*islice(random_walk(),10))

我尝试了以下代码来可视化在长度为 1000 的随机游走中访问不同单元格的频率：

import matplotlib.pyplot as plt; plt.imshow([[1,2,4],[0,3,-1]],cmap = "Blues"); plt.colorbar(); plt.show()
import itertools
import numpy as np

# generate random walk of length 1000
walk = list(itertools.islice(random_walk(),1000))

xy = list(zip(*walk))

x = list(xy[0])
y = list(xy[1])

heatmap,xedges,yedges = np.histogram2d(x,y,bins=[np.arange(min(x),max(x),np.arange(min(y),max(y),1)])
extent = [xedges[0],xedges[-1],yedges[0],yedges[-1]]
plt.imshow(heatmap,origin='lower',extent=extent)

但它没有绘制正确的图像，有没有人有想法将不同单元格在长度为 1000 的随机游走中访问的频率可视化？

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）

imshowmatplotlibpythonrandom-walk