import numpy as np
from matplotlib import pyplot as plt
%matplotlib inline
plt.rcParams['figure.figsize'] = 10, 10
data = np.array(
((1, 2, 1),
(3, 3, 0),
(0, 0, 4))
)
plt.imshow(
data,
interpolation='nearest',
aspect='auto',
cmap='Reds', # other colors: https://matplotlib.org/examples/color/colormaps_reference.html
vmin=0,
vmax=5,
)
plt.colorbar()
<matplotlib.colorbar.Colorbar at 0x7f15b874af60>