如何使用colorbar在python中绘制2d矩阵？(如matlab中的imagesc)

Question

在Matlab中,我可以data很容易地用矩阵可视化矩阵

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

现在我想在python中做同样的事情.我已经知道如何绘制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()

Answer 2

另一种可能性是使用plt.matshow（）

import numpy as np
import matplotlib.pyplot as plt

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