import matplotlib.pyplot as plt
import matplotlib as mpl
colors = ['white', 'blue', 'cyan', 'Lime','yellow']
cmap = mpl.colors.ListedColormap(colors)
plt.imshow(a1[num1, :, :], cmap=cmap)
import matplotlib.pyplot as plt
import matplotlib as mpl
colors = ['white', 'blue', 'cyan', 'Lime','yellow']
cmap = mpl.colors.ListedColormap(colors)
plt.imshow(a1[num1, :, :], cmap=cmap)