如何用彩條在python中繪製2d矩陣？ （如在Matlab於imagesc）

Question

在Matlab中，我可以很容易地與

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

現在我想要做同樣的事情在python可視化矩陣data。我已經知道如何繪製一個2D矩陣（numpy數組）：

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

但我不知道如何給它添加一個顏色條。有任何想法嗎？

Answer 1

import numpy as np 
import matplotlib.pyplot as plt 

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