def draw_heatmap(xedges, yedges, heatmap): extent = [xedges[0], xedges[-1], yedges[0], yedges[-1]] plt.figure(figsize=(8, 8)) plt.imshow(heatmap, extent=extent) plt.show()