如何保存从numpy数组创建的图像？

Question

我想使用plt.savefig（）保存从numpy arry创建的图像，但它失败了。我的图片是空白的。我的代码如下。

import matplotlib.pyplot as plt
import numpy as np

array = np.arange(10000).reshape((100,100))
plt.savefig('test.png', dpi = 300)
plt.imshow(array)

如何在我的图片中添加彩条？非常感谢你。 就像下图一样。

Answer 1

fig = plt.figure(figsize = (12,12))

data = np.arange(10000).reshape((100,100))
plt.subplot(111)
plt.imshow(data,interpolation='nearest', cmap='bone', origin='lower')
plt.colorbar(shrink=.92)

plt.savefig('imshow_ex.png', dpi=100)
plt.show()

上面的代码可以解决我的问题。谢谢。