import numpy as np
import matplotlib.pyplot as plt
x = np.linspace(-3,3,400)
y = np.linspace(-3,3,400)
X,Y = np.meshgrid(x,y)
Z = np.sin(X**2 + Y**2)
plt.imshow(Z, cmap='plasma')
plt.show()