Tengo una lista barColors llena de valores RGB que estoy tratando de trazar:

[(120.0304, 117.9008, 122.6944),
 (66.3952, 65.0592, 69.088),
 (22.2944, 24.3504, 26.5872),
 (22.5744, 24.8352, 26.9152),
 (0.0, 0.0, 0.0),
 (94.864, 81.6416, 67.2272),
 (92.5328, 79.6288, 66.2928),
 (102.104, 86.7856, 71.3408),
 (77.664, 65.0288, 52.712),
 (78.2688, 69.3488, 60.1936),
 (19.0432, 17.696, 17.7792),
 (20.0432, 22.4064, 27.5456),
 (30.7776, 32.4288, 36.5024),
 (46.192, 49.8928, 54.7008),
 (45.8016, 48.328, 55.1968),
.
.
.

Me gustaría crear una imagen con cortes verticales para cada color (fila) en la lista barColors.

He intentado:

title = "p"
#creating bar image
barImg = Image.new("RGB",(len(barColors), max([1,int(len(barColors)/2.5)])))

#adding bars to the image
barFullData = [x for x in barColors] * barImg.size[1]
barImg.putdata(barFullData)

#folder to store bar images
if not os.path.isdir("bars"):
    os.mkdir("bars")


#saving image
barImg.save("bars/{}_{}.png".format(title,method))
barImg.show()

Pero recibo un error:

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-42-5a3d9f5e3a90> in <module>()
      5 #adding bars to the image
      6 barFullData = [x for x in barColors] * barImg.size[1]
----> 7 barImg.putdata(barFullData)
      8 
      9 #folder to store bar images

C:\Users\meezy\Anaconda3\lib\site-packages\PIL\Image.py in putdata(self, data, scale, offset)
   1456             self._copy()
   1457 
-> 1458         self.im.putdata(data, scale, offset)
   1459 
   1460     def putpalette(self, data, rawmode="RGB"):

TypeError: integer argument expected, got float
3
Matt M 10 may. 2019 a las 04:25

3 respuestas

La mejor respuesta

Las imágenes pil deben contener enteros. Recomendaría usar PIL y numpy de la siguiente manera:

import numpy as np
from PIL import Image

barColors = [(120.0304, 117.9008, 122.6944),
             (66.3952, 65.0592, 69.088),
             (22.2944, 24.3504, 26.5872),
             (22.5744, 24.8352, 26.9152),
             (0.0, 0.0, 0.0),
             (94.864, 81.6416, 67.2272),
             (92.5328, 79.6288, 66.2928),
             (102.104, 86.7856, 71.3408),
             (77.664, 65.0288, 52.712),
             (78.2688, 69.3488, 60.1936),
             (19.0432, 17.696, 17.7792),
             (20.0432, 22.4064, 27.5456),
             (30.7776, 32.4288, 36.5024),
             (46.192, 49.8928, 54.7008),
             (45.8016, 48.328, 55.1968)]

barColors = (np.array(barColors)).astype(np.uint8)

title = "p"
#creating bar image
cols = len(barColors)
rows = max([1,int(cols/2.5)])

# Create color Array
barFullData = np.tile(barColors, (rows,1)).reshape(rows, cols, 3)
# Create Image from Array
barImg = Image.fromarray(barFullData, 'RGB')

#saving image
barImg.save("{}_{}.png".format(title,"method"))
barImg.show()

enter image description here

Versión ampliada:

enter image description here

1
ImportanceOfBeingErnest 10 may. 2019 a las 02:33

Puede usar axvspan para trazar una barra por color:

from matplotlib import pyplot as plt

scaled_colours = [[color / 255 for color in row] for row in colours]

fig, ax = plt.subplots(figsize=(6, 6))

ax.axis(xmin=0, xmax=len(scaled_colours))
ax.tick_params(left=False, labelleft=False, bottom=False, labelbottom=False)

for index, colour in enumerate(scaled_colours):
    ax.axvspan(index, index + 1, color=colour)

Tenga en cuenta que si sus valores de color originales son floats, deben reducirse al rango [0, 1].

Salida:

enter image description here

Luego puede guardar el resultado con fig.savefig.

1
gmds 10 may. 2019 a las 01:38

Además de las respuestas dadas, otra opción es usar palplot de seaborn.

import seaborn as sns
colors = np.array([
 (120.0304, 117.9008, 122.6944),
 (66.3952, 65.0592, 69.088),
 (22.2944, 24.3504, 26.5872),
 (22.5744, 24.8352, 26.9152),
 (0.0, 0.0, 0.0),
 (94.864, 81.6416, 67.2272),
 (92.5328, 79.6288, 66.2928),
 (102.104, 86.7856, 71.3408),
 (77.664, 65.0288, 52.712),
 (78.2688, 69.3488, 60.1936),
 (19.0432, 17.696, 17.7792),
 (20.0432, 22.4064, 27.5456),
 (30.7776, 32.4288, 36.5024),
 (46.192, 49.8928, 54.7008),
 (45.8016, 48.328, 55.1968)]) / 255.0

sns.palplot(colors)

barras verticales de colores

Obtuve la imagen después de arrastrar la ventana de trazado al ancho y alto apropiados. palplot tiene un parámetro size que controla la altura. Salvar la figura usando fig = plt.gcf() y luego fig.savefig o directamente usando plt.savefig()

Espero que esto ayude.

1
Prasanth 10 may. 2019 a las 08:47