Как сохранить данные гистограммы в файле CSV в Python?

Question

Из приведенного ниже кода Python я могу построить гистограмму трех цветовых компонентов цвета изображение .Но я хотел бы сохранить эти данные гистограммы в файл CSV, чтобы получить эту гистограмму в будущем.Как я могу это сделать?

from PIL import Image
import matplotlib.pyplot as plt

def getRed(redVal):
    return '#%02x%02x%02x' % (redVal, 0, 0)

def getGreen(greenVal):
    return '#%02x%02x%02x' % (0, greenVal, 0)

def getBlue(blueVal):
    return '#%02x%02x%02x' % (0, 0, blueVal)

# Create an Image with specific RGB value
image = Image.open("baboon.ppm")
# Modify the color of two pixels
image.putpixel((0,1), (1,1,5))
image.putpixel((0,2), (2,1,5))

# Display the image
#image.show()
# Get the color histogram of the image
histogram = image.histogram()
# Take only the Red counts
l1 = histogram[0:256]
# Take only the Blue counts
l2 = histogram[256:512]
# Take only the Green counts
l3 = histogram[512:768]
plt.figure(0)
# R histogram
for i in range(0, 256):
    plt.bar(i, l1[i], color = getRed(i), edgecolor=getRed(i), alpha=0.3)

# G histogram
plt.figure(1)
for i in range(0, 256):
    plt.bar(i, l2[i], color = getGreen(i), edgecolor=getGreen(i),alpha=0.3)


# B histogram
plt.figure(2)
for i in range(0, 256):
    plt.bar(i, l3[i], color = getBlue(i), edgecolor=getBlue(i),alpha=0.3)

plt.show()

Вот вывод для красного канала гистограмма

Answer 1

l1, l2 и l3 - это просто списки.
Вы можете преобразовать их в одну строку и записать строку в простой текстовый файл, например:

def writeCsv( RList, GList, BList ):
    outString = '\r\n'.join([
        ';'.join( map( str, RList ) ),
        ';'.join( map( str, GList ) ),
        ';'.join( map( str, BList ) )
    ])
    print( outString )
    f = open( 'csv_file.csv', 'wb' )
    f.write( outString )
    f.close()

def loadCsv():
    f = open( 'csv_file.csv', 'rb' )
    out = f.read()
    f.close()
    out = out.split('\r\n');
    out = [ x.split(';') for x in out ]
    return out

writeCsv( l1, l2, l3 )
print('--')
print( loadCsv() )

Может быть желательно сохранить файлкак «RGB, RGB, ...» вместо «RRR ..., GGG ..., BBB ...»:

def writeCsvRGB( RList, GList, BList ):
    temp = zip( RList, GList, BList )
    temp = [ ';'.join( map( str, x ) ) for x in temp ]
    temp  = '\r\n'.join( temp )
    print( temp )

    f = open( 'csv_file.csv', 'wb' )
    f.write( temp )
    f.close()