媒体大数据 色彩提取

代码：

import numpy as np
import os
from PIL import Image
from scipy.cluster.vq import vq, kmeans, whiten
import matplotlib.pyplot as plt
def colordata(filename,n=3):
    img=Image.open(filename)
    img=img.rotate(0)
    img.thumbnail((200,200))
    w,h=img.size
    print(w,h) 
    print('w*h=',w*h)
    plt.axis('off')
    plt.imshow(img)
    plt.show()
    points=[]
    for count,color in img.getcolors(w*h):
        points.append(color)
    return points
def kmeansFun(data,n):
    
    data=np.array(data,dtype=float)  #聚类需要是Float或者Double
    center_init = np.array(data[0:n])   #初始聚类中心
    print(type(center_init))

    print(type(center_init))   
    print("Center_init: \n",center_init)
   
    centers, loss = kmeans(data, n) #   n是聚类中心个数
    # 可以写kmeans(wf,2)， 2表示两个质心，同时启用iter参数
    print("Centers:", centers)
    print("Loss: ", loss)

    plt.scatter(data[:,0], data[:,1], c='g')
    plt.scatter(centers[:, 0], centers[:, 1], c='r')
    plt.show()
os.chdir("F:\\图片\\")
imgdata=colordata("胡歌.jpg")
kmeansFun(imgdata,5)

效果