def myplot(x, y, nb=32, xsize=500, ysize=500): heatmap, xedges, yedges = np.histogram2d(x, y, bins=nb) extent = [xedges[0], xedges[-1], yedges[0], yedges[-1]] return heatmap.T, extent