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

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

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

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 的随机游走中访问的频率可视化？