# https://www.globalsino.com/ICs/
# Heatmap with getting rid of the zero values, e.g. at the corners


import matplotlib.pyplot as plt
import numpy as np

data = np.array([[ 0. ,  0. ,  1. ,  0. ,  1. ,  0. , 0.],
       [ 0. ,  2. ,  1. ,  4. ,  0. ,  1., 0.],
       [ 0. ,  1. ,  0.3,  3. ,  0. ,  0., 1.],
       [ 0. ,  1. ,  0. ,  0.6,  1. ,  4., 2.],
       [ 0. ,  1. ,  1. ,  1. ,  1. ,  0., 0.2],
       [ 0. ,  0. ,  1. ,  1. ,  0. ,  0., 0.],
       [ 0. ,  0. ,  0. ,  0. ,  2. ,  0., 0. ]])

data[data == 0] = np.nan
plt.figure(1)
plt.imshow(data ,interpolation='none', vmin = 0)
plt.colorbar()
plt.show()