plt.imshow(
    output,
    cmap="gray",
    extent=[np.min(x), np.max(x), np.min(y), np.max(y)]
)