You want the extent
kwarg
ax.imshow(data, extent=[0, 1, 0, 1])
See Imshow: extent and aspect for a more detailed example.
You want the extent
kwarg
ax.imshow(data, extent=[0, 1, 0, 1])
See Imshow: extent and aspect for a more detailed example.