Comment tracer une matrice 2D en python avec colorbar? (comme imagesc dans Matlab)

Question

Dans Matlab, je peux visualiser une matrice data assez facilement avec

data = Rand(10,10) % Createas a 10 x 10 random matrix
imagesc(data);
colorbar;

Maintenant, je veux faire la même chose en python. Je sais déjà comment tracer une matrice 2D (tableau numpy):

from matplotlib.pyplot import imshow
import numpy as np
data = np.random.random((10,10))
imshow(np.asarray(img))

mais je ne sais pas comment ajouter une barre de couleur. Des idées?

Answer 1

import numpy as np
import matplotlib.pyplot as plt

plt.imshow(np.random.random((50,50)));
plt.colorbar()
plt.show()

Answer 2

Une autre possibilité consiste à utiliser plt.matshow ()

import numpy as np
import matplotlib.pyplot as plt

plt.matshow(np.random.random((50,50)));
plt.colorbar()
plt.show()

https://matplotlib.org/api/_as_gen/matplotlib.pyplot.matshow.html