文章目录
今天为大家分享一个有趣的可视化技巧,如何从图片中提取颜色然后绘制成可视化图表,如下图所示
在示例照片当中有着各种各样的颜色,我们将通过Python
中的可视化模块以及opencv
模块来识别出图片当中所有的颜色要素,并且将其添加到可视化图表的配色当中
导入模块并加载图片
那么按照惯例,第一步一般都是导入模块,可视化用到的模块是matplotlib
模块,我们将图片中的颜色抽取出来之后会保存在颜色映射表中,所以要使用到colormap
模块,同样也需要导入进来
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import matplotlib.patches as patches
import matplotlib.image as mpimg
from PIL import Image
from matplotlib.offsetbox import OffsetImage, AnnotationBbox
import cv2
import extcolors
from colormap import rgb2hex
然后我们先来加载一下图片,代码如下
input\_name = 'test\_1.png'
img = plt.imread(input\_name)
plt.imshow(img)
plt.axis('off')
plt.show()
output
提取颜色并整合成表格
我们调用的是extcolors
模块来从图片中提取颜色,输出的结果是RGB
形式呈现出来的颜色,代码如下
colors\_x = extcolors.extract\_from\_path(img\_url, tolerance=12, limit = 12)
colors\_x
output
([((3, 107, 144), 180316),
((17, 129, 140), 139930),
((89, 126, 118), 134080),
((125, 148, 154), 20636),
((63, 112, 126), 18728),
((207, 220, 226), 11037),
((255, 255, 255), 7496),
((28, 80, 117), 4972),
((166, 191, 198), 4327),
((60, 150, 140), 4197),
((90, 94, 59), 3313),
((56, 66, 39), 1669)],
538200)
我们将上述的结果整合成一个DataFrame
数据集,代码如下
def color\_to\_df(input\_color):
colors\_pre\_list = str(input\_color).replace('(\[(', '').split(', ('