2D配列を色付きの.png形式で保存する

Question

このPythonプログラムを作成して、行列（2D配列）を作成して.pngファイルに保存しました。プログラムはコンパイルされ、エラーなく実行されます。 IMAGE.pngファイルも作成されますが、pngファイルは開かれません。 MSPaintで開こうとすると、次のように表示されます。

イメージを開けません。有効なビットマップファイルまたはその形式は現在サポートされていません。

私の目的は、2D配列に格納された数字に基づいてRBG png画像を作成することです。

ソースコード：エラーメッセージがないように私が間違っていたところ、私は理解していない

import numpy; 
    import png; 

    imagearray = numpy.zeros(shape=(512,512)); 

    /* Code to insert one '1', '2', '3' in certain locations 
     of the numpy 2D Array. Rest of the location by default stores zero '0'.*/ 


    f = open("IMAGE.png", 'wb'); 
    f.write(imagearray); 
    f.close(); 

。助けてください。

PS-単に画像ファイルとして保存したいので、Python2.7でもっと簡単に行える方法があればお勧めします。

Answer 1

画像をmatplotlibでプロットして保存します。

import matplotlib.pyplot as plt 
import numpy as np 
a = np.random.uniform(size=(25, 25, 3)) # random 3D array 
plt.imshow(a) 
plt.savefig("img.png") 

imshow()this questionごとに、軸や空白を削除するには、各種の中interpolation（examples here）関心のパラメータ、およびcmap（カラーマップ）

があります

plt.axis('off') 
plt.savefig("img.png", bbox_inches='tight')