如何使用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()

here