将多个大小为 (22,6,2840) 的 3D 数组保存为图像

Question

我正在生成多个大小为 (22,6,2840) 的 3D numpy 数组，每个数组包含 22 个大小为 (6,2840) 的数组。现在我想将此数组 (22,6,2840) 保存为图像。我不知道我能不能做到。我尝试使用plt.savefig 执行此操作，但没有成功。我正在尝试超过 2 周来寻找我能做到的事情。 任何帮助将不胜感激。

    signals=np.zeros((22,6,2840))
    t=0
    movement=int(S*256)
    if(S==0):
        movement=_SIZE_WINDOW_SPECTOGRAM        
    while data.shape[1]-(t*movement+_SIZE_WINDOW_SPECTOGRAM) > 0:
        for i in range(0, 22):
            start = t*movement
            stop = start+_SIZE_WINDOW_SPECTOGRAM
            signals[i,:]=wavelet(data[i,start:stop])
        if(signalsBlock is None):
            signalsBlock=np.array([signals])
        else:
            signalsBlock=np.append(signalsBlock, [signals], axis=0)
        nSpectogram=nSpectogram+1
        if(signalsBlock.shape[0]==50):
            saveSignalsOnDisk(signalsBlock, nSpectogram)
            signalsBlock=None
              
        t = t+1

Answer 1

尝试使用PyPNG 库。您必须将数组重塑为二维格式，然后将其写为 png。图书馆的链接是here

    image_2d = numpy.reshape(image_3d, (-1, column_count * plane_count))
    pngWriter.write(out, image_2d)

此外，here 还提供了另一种使用 PIL Image 的方法。但是，这主要适用于 RGB 样式的 3 通道图像。