ggzhang0071 / graphPartitioning

0 stars 0 forks source link

从图片找到其颜色用于画图 python code #4

Open ggzhang0071 opened 4 years ago

ggzhang0071 commented 4 years ago
from PIL import Image
im=Image.open("beatutulFig.png")
pix=im.load()

width=im.size[0]
height=im.size[1]
Colors=[]
for x in range(width):
    for y in range(height):
        r,g,b=pix[x,y]
        Colors.append((r,g,b))

from  collections import Counter
N=8
BeatutifulColors=[]
for i,item in enumerate(Counter(Colors).most_common(N)[1:]):
    BeatutifulColors.append((item[0][0]/255,item[0][1]/255,item[0][2]/255,1))  #rgba 是[0,1] 之间的
ggzhang0071 commented 4 years ago

主要是使用 Counter 计算 然后 转换成成 rgba