[解決済み】numpyの配列を画像に変換する（表示する）にはどうしたらいいですか？

質問

このような配列を作成しました。

import numpy as np
data = np.zeros( (512,512,3), dtype=np.uint8)
data[256,256] = [255,0,0]

512x512の画像の中央に赤い点を1つだけ表示させたいのです。(少なくとも最初は...そこから残りを理解できると思います)

どのように解決するのですか？

PILを使って、画像を作成（表示）することができます。

from PIL import Image
import numpy as np

w, h = 512, 512
data = np.zeros((h, w, 3), dtype=np.uint8)
data[0:256, 0:256] = [255, 0, 0] # red patch in upper left
img = Image.fromarray(data, 'RGB')
img.save('my.png')
img.show()