Функция imshow()
с параметрами interpolation='nearest'
и cmap='hot'
должна делать то, что вы хотите.
import matplotlib.pyplot as plt
import numpy as np
a = np.random.random((16, 16))
plt.imshow(a, cmap='hot', interpolation='nearest')
plt.show()