将 Numpy 数组保存为图像（说明）

我在之前的帖子中找到了答案：将 Numpy 数组保存为图像 https://stackoverflow.com/questions/902761/saving-a-numpy-array-as-an-image。唯一的问题是，没有太多关于使用 PyPNG 模块的说明。

网上只有几个例子——http://packages.python.org/pypng/ex.html#numpy http://packages.python.org/pypng/ex.html#numpy http://nullege.com/codes/search/png.Writer.write http://nullege.com/codes/search/png.Writer.write

但是，鉴于这样的 .write 错误，我该怎么办：

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/Library/Frameworks/Python.framework/Versions/2.7/lib/python2.7/site-packages/png.py", line 638, in write
    nrows = self.write_passes(outfile, rows)
  File "/Library/Frameworks/Python.framework/Versions/2.7/lib/python2.7/site-packages/png.py", line 783, in write_passes
    extend(row)
  File "/Library/Frameworks/Python.framework/Versions/2.7/lib/python2.7/site-packages/png.py", line 780, in <lambda>
    return lambda sl: f(map(int, sl))
TypeError: argument 2 to map() must support iteration

这是我的代码 PCA_tool.py 中发生错误的位置（错误出现在“folder.write(outfilename, PrimaryComponent”之后）：

#PrincipalComponent.save(path+'transform_'+str(each)+'.png', format='PNG')
outfilename = open(str(path)+'transformed/transform_'+str(each)+'.png', 'wb')
folder = png.Writer(m,n,greyscale=True)
folder.write(outfilename, PrincipalComponent)
outfilename.close()

sys.exit(0)

我正在尝试将 8400 元素 numpy.ndarray 保存为 n=80 列、m=105 行灰度 png 图像。

Thanks,

使用 PIL 可能会更好：

from PIL import Image
import numpy as np

data = np.random.random((100,100))

#Rescale to 0-255 and convert to uint8
rescaled = (255.0 / data.max() * (data - data.min())).astype(np.uint8)

im = Image.fromarray(rescaled)
im.save('test.png')