NumPy: преобразование массива в изображение

NumPy: Array Object Exercise-109 с решением

Напишите программу NumPy для преобразования массива numy в изображение. Покажите изображение.

Пример решения :

Код Python:

from PIL import Image
import numpy as np
img_w, img_h = 200, 200
data = np.zeros((img_h, img_w, 3), dtype=np.uint8)
data[100, 100] = [255, 0, 0]
img = Image.fromarray(data, 'RGB')
img.save('test.png')
img.show()

Пример вывода:

Редактор кода Python:

Есть другой способ решить это решение? Внесите свой код (и комментарии) через Disqus.

Предыдущий: Напишите программу NumPy для преобразования изображения PIL в массив numpy.
Далее: Напишите программу NumPy для удаления значений nan из заданного массива.