将具有像素值的csv文件转换为图像

如果最后一列中有名称，则必须在没有last元素的情况下进行转换data[:-1]。并使用filenamename中的最后一列savefig( data[-1] + str(i) + '.jpg' )。没有扩展，它可能不知道要写入哪种类型的图像。

您必须分别计算i每个标签-即 使用字典

i = dict() 
i[label] = 0 

# later 

i[label] += 1 
savefig( label + str(i[label]) + '.jpg' 

您也可以使用PIL/'pillow instead ofmatplotlib`编写它。

from PIL import Image    

image = Image.fromarray(pixels)
image.save(filename)

import numpy as np
import csv
from PIL import Image    

counter = dict()

with open('myfile.csv') as csv_file:
    csv_reader = csv.reader(csv_file)

    # skip headers
    next(csv_reader)

    for row in csv.reader(csv_reader):

        pixels = row[:-1] # without label
        pixels = np.array(pixels, dtype='uint8')
        pixels = pixels.reshape((28, 28))
        image = Image.fromarray(pixels)

        label = row[-1]

        if label not in counter:
            counter[label] = 0
        counter[label] += 1

        filename = '{}{}.jpg'.format(label, counter[label])
        image.save(filename)

        print('saved:', filename)

编辑：示例，其中显示了一行数据，以便那些无需下载csv文件即可对其进行测试的人。

import numpy as np
import csv
from PIL import Image

counter = dict()

row = [
        255,   0,   0,   0,   0,   0,  255,
          0, 255, 255, 255, 255, 255,    0,
          0, 255,   0, 255,   0, 255,    0,
          0, 255, 255, 255, 255, 255,    0,
          0, 255,   0,   0,   0, 255,    0,
          0, 255, 255, 255, 255, 255,    0,
        255,   0,   0,   0,   0,   0,  255,
        'face'
      ]

pixels = row[:-1]

pixels = np.array(pixels, dtype='uint8')
pixels = pixels.reshape((7, 7))
image = Image.fromarray(pixels)

label = row[-1]

if label not in counter:
    counter[label] = 0
counter[label] += 1

filename = '{}{}.png'.format(label, counter[label])

image.save(filename)

print('saved:', filename)

结果face1.png：

编辑：我检查了您的csv文件，像素不是整数值而是浮点（正，负）值，所以您不能使用uint8。它必须是float。

pixels = np.array(pixels, dtype='float')

您可以将图像转换为RGB或灰度L以保存它

image = image.convert('RGB')
image = image.convert('L')

但似乎无法转换负值。

使用

plt.imsave(filename, pixels)

我得到了预期的结果

import numpy as np
import csv
from PIL import Image    
import matplotlib.pyplot as plt

counter = dict()

with open('gen_image_wgan.csv') as csv_file:
    csv_reader = csv.reader(csv_file)

    # skip headers
    next(csv_reader)

    for row in csv_reader:

        pixels = row[:-1] # without label
        pixels = np.array(pixels, dtype='float')
        pixels = pixels.reshape((28, 28))

        label = row[-1]

        if label not in counter:
            counter[label] = 0
        counter[label] += 1

        filename = '{}{}.png'.format(label, counter[label])
        plt.imsave(filename, pixels)

        print('saved:', filename)