Python實(shí)現(xiàn)提取圖片中顏色并繪制成可視化圖表
今天小編來(lái)為大家分享一個(gè)有趣的可視化技巧,如何從圖片中提取顏色然后繪制成可視化圖表,如下圖所示

在示例照片當(dāng)中有著各種各樣的顏色,我們將通過(guò)Python中的可視化模塊以及opencv模塊來(lái)識(shí)別出圖片當(dāng)中所有的顏色要素,并且將其添加到可視化圖表的配色當(dāng)中
導(dǎo)入模塊并加載圖片
那么按照慣例,第一步一般都是導(dǎo)入模塊,可視化用到的模塊是matplotlib模塊,我們將圖片中的顏色抽取出來(lái)之后會(huì)保存在顏色映射表中,所以要使用到colormap模塊,同樣也需要導(dǎo)入進(jìn)來(lái)
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
然后我們先來(lái)加載一下圖片,代碼如下
input_name?=?'test_1.png'
img?=?plt.imread(input_name)
plt.imshow(img)
plt.axis('off')
plt.show()output

提取顏色并整合成表格
我們調(diào)用的是extcolors模塊來(lái)從圖片中提取顏色,輸出的結(jié)果是RGB形式呈現(xiàn)出來(lái)的顏色,代碼如下
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)
我們將上述的結(jié)果整合成一個(gè)DataFrame數(shù)據(jù)集,代碼如下
def?color_to_df(input_color):
????colors_pre_list?=?str(input_color).replace('([(',?'').split(',?(')[0:-1]
????df_rgb?=?[i.split('),?')[0]?+?')'?for?i?in?colors_pre_list]
????df_percent?=?[i.split('),?')[1].replace(')',?'')?for?i?in?colors_pre_list]
????#?將RGB轉(zhuǎn)換成十六進(jìn)制的顏色
????df_color_up?=?[rgb2hex(int(i.split(",?")[0].replace("(",?"")),
???????????????????????????int(i.split(",?")[1]),
???????????????????????????int(i.split(",?")[2].replace(")",?"")))?for?i?in?df_rgb]
????df?=?pd.DataFrame(zip(df_color_up,?df_percent),?columns=['c_code',?'occurence'])
????return?df我們嘗試調(diào)用上面我們自定義的函數(shù),輸出的結(jié)果至DataFrame數(shù)據(jù)集當(dāng)中
df_color?=?color_to_df(colors_x) df_color
output

繪制圖表
接下來(lái)便是繪制圖表的階段了,用到的是matplotlib模塊,代碼如下
fig,?ax?=?plt.subplots(figsize=(90,90),dpi=10)
wedges,?text?=?ax.pie(list_precent,
??????????????????????labels=?text_c,
??????????????????????labeldistance=?1.05,
??????????????????????colors?=?list_color,
??????????????????????textprops={'fontsize':?120,?'color':'black'}
?????????????????????)
plt.setp(wedges,?width=0.3)
ax.set_aspect("equal")
fig.set_facecolor('white')
plt.show()output

從出來(lái)的餅圖中顯示了每種不同顏色的占比,我們更進(jìn)一步將原圖放置在圓環(huán)當(dāng)中,
imagebox?=?OffsetImage(img,?zoom=2.3) ab?=?AnnotationBbox(imagebox,?(0,?0)) ax1.add_artist(ab)
output

最后制作一張調(diào)色盤(pán),將原圖中的各種不同顏色都羅列開(kāi)來(lái),代碼如下
##?調(diào)色盤(pán)
x_posi,?y_posi,?y_posi2?=?160,?-170,?-170
for?c?in?list_color:
????if?list_color.index(c)?<=?5:
????????y_posi?+=?180
????????rect?=?patches.Rectangle((x_posi,?y_posi),?360,?160,?facecolor?=?c)
????????ax2.add_patch(rect)
????????ax2.text(x?=?x_posi+400,?y?=?y_posi+100,?s?=?c,?fontdict={'fontsize':?190})
????else:
????????y_posi2?+=?180
????????rect?=?patches.Rectangle((x_posi?+?1000,?y_posi2),?360,?160,?facecolor?=?c)
????????ax2.add_artist(rect)
????????ax2.text(x?=?x_posi+1400,?y?=?y_posi2+100,?s?=?c,?fontdict={'fontsize':?190})
ax2.axis('off')
fig.set_facecolor('white')
plt.imshow(bg)???????
plt.tight_layout()output

實(shí)戰(zhàn)環(huán)節(jié)
這一塊兒是實(shí)戰(zhàn)環(huán)節(jié),我們將上述所有的代碼封裝成一個(gè)完整的函數(shù)
def?exact_color(input_image,?resize,?tolerance,?zoom):
????
????output_width?=?resize
????img?=?Image.open(input_image)
????if?img.size[0]?>=?resize:
????????wpercent?=?(output_width/float(img.size[0]))
????????hsize?=?int((float(img.size[1])*float(wpercent)))
????????img?=?img.resize((output_width,hsize),?Image.ANTIALIAS)
????????resize_name?=?'resize_'+?input_image
????????img.save(resize_name)
????else:
????????resize_name?=?input_image
????
????fig.set_facecolor('white')
????ax2.axis('off')
????bg?=?plt.imread('bg.png')
????plt.imshow(bg)???????
????plt.tight_layout()
????return?plt.show()
????
exact_color('test_2.png',?900,?12,?2.5)output

以上就是Python實(shí)現(xiàn)提取圖片中顏色并繪制成可視化圖表的詳細(xì)內(nèi)容,更多關(guān)于Python提取圖片顏色的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章!
相關(guān)文章
Pytorch中torch.stack()函數(shù)的深入解析
在pytorch中常見(jiàn)的拼接函數(shù)主要是兩個(gè),分別是:stack()和cat(),下面這篇文章主要給大家介紹了關(guān)于Pytorch中torch.stack()函數(shù)的相關(guān)資料,文中通過(guò)實(shí)例代碼介紹的非常詳細(xì),需要的朋友可以參考下2022-08-08
用pycharm開(kāi)發(fā)django項(xiàng)目示例代碼
這篇文章主要介紹了用pycharm開(kāi)發(fā)django項(xiàng)目示例,小編覺(jué)得挺不錯(cuò)的,現(xiàn)在分享給大家,也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2018-10-10
Python中使用kitti數(shù)據(jù)集實(shí)現(xiàn)自動(dòng)駕駛(繪制出所有物體的行駛軌跡)
這篇文章主要介紹了Python中使用kitti數(shù)據(jù)集實(shí)現(xiàn)自動(dòng)駕駛——繪制出所有物體的行駛軌跡,本次內(nèi)容主要是畫(huà)出kitti車(chē)的行駛的軌跡,需要的朋友可以參考下2022-06-06
Python Sqlalchemy如何實(shí)現(xiàn)select for update
這篇文章主要介紹了Python Sqlalchemy如何實(shí)現(xiàn)select for update,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-10-10
Python3爬蟲(chóng)里關(guān)于Splash負(fù)載均衡配置詳解
在本篇文章里小編給大家分享了關(guān)于Python3爬蟲(chóng)里關(guān)于Splash負(fù)載均衡配置的相關(guān)內(nèi)容,需要的朋友們可以學(xué)習(xí)參考下。2020-07-07
基于Python數(shù)據(jù)可視化利器Matplotlib,繪圖入門(mén)篇,Pyplot詳解
下面小編就為大家?guī)?lái)一篇基于Python數(shù)據(jù)可視化利器Matplotlib,繪圖入門(mén)篇,Pyplot詳解。小編覺(jué)得挺不錯(cuò)的,現(xiàn)在就分享給大家,也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2017-10-10
Python學(xué)習(xí)思維導(dǎo)圖(必看篇)
下面小編就為大家?guī)?lái)一篇Python學(xué)習(xí)思維導(dǎo)圖(必看篇)。小編覺(jué)得挺不錯(cuò)的,現(xiàn)在就分享給大家,也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2017-06-06
PyQt5執(zhí)行耗時(shí)操作導(dǎo)致界面卡死或未響應(yīng)的原因及解決辦法
這篇文章主要給大家介紹了關(guān)于PyQt5執(zhí)行耗時(shí)操作導(dǎo)致界面卡死或未響應(yīng)的原因及解決辦法,由于耗時(shí)的操作會(huì)獨(dú)占系統(tǒng)cpu資源,讓界面卡死在那里,文中通過(guò)代碼介紹的非常詳細(xì),需要的朋友可以參考下2023-12-12
Python tempfile模塊學(xué)習(xí)筆記(臨時(shí)文件)
這篇文章主要介紹了Python tempfile模塊學(xué)習(xí)筆記,著重講解了模塊下的幾個(gè)函數(shù),需要的朋友可以參考下2014-05-05

