Матрица Numpy в образ

Question

У меня есть 2d массив со значением int, которое я хочу преобразовать в изображение.

2d массив генерируется случайным образом между 1-3, с учетом того, что соседние int это в массиве, я хочу преобразовать 1,2,3 в R,G,B в изображении, чтобы лучше увидеть, каков результат генератора является.

Каков наилучший способ сделать это?

Answer 1

Я бы использовал библиотеку http://matplotlib.org/. Просто используйте plt.imshow или же plt.pcolormesh (второй технически лучше для дискретных значений). В этом случае цветовая карта по умолчанию довольно близка к RGB, но вы можете использовать другую цветовую карту, если хотите. Например:

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Вам может понадобиться позвонить plt.show() чтобы заставить его рисовать, если вы не используете IPython.