Numpy Matrix成图像

Question

我有一个带有int值的二维数组，我希望将其转换为图像。

2d数组是在1-3之间随机生成的，考虑到它在数组中的相邻int，我想将1,2,3转换为图像中的R，G，B，以便更好地看到结果发电机是。

这样做的最佳方式是什么？

Answer 1

我会使用matplotlib库。只需使用plt.imshow或plt.pcolormesh（第二个在技术上更适合离散值）。在这种情况下，默认的色彩映射非常接近RGB，但如果您愿意，可以使用another colormap。例如：

import numpy as np
import matplotlib.pyplot as plt

# Creating random 1-3 data in a 2D array
data = np.random.randint(1,4,[100,150])

plt.pcolormesh(a)

我正在使用IPython和%matplotlib inline，如果您不使用IPython，可能需要调用plt.show()来绘制它。