import random
from sklearn import datasets
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import cv2
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
from sklearn.cluster import KMeans
from sklearn.preprocessing import StandardScaler
from mpl_toolkits.mplot3d import Axes3D
def ComMatrixRGB(f,filename=""):
X = []
for i,v1 in enumerate(f):
for j,v2 in enumerate(v1):
#print(j,v2)
X.append(list([int(v2[0]),int(v2[1]),int(v2[2])]))
return X
def main():
f1 = cv2.imread("11.png")
data = np.array(ComMatrixRGB(f1)[:])
#data = np.resize(f1,(f1.shape[1]*f1.shape[0],f1.shape[2]))可以替换上一行
#标准化
transfer = StandardScaler()
data_new = transfer.fit_transform(data)
data_new
#预估计流程
estimator = KMeans(n_clusters = 6)
estimator.fit(data_new)
y_pred = estimator.predict(data_new)
#也可以不预测
#cluster = KMeans(n_clusters = 9).fit(data_new)
#y_pred = cluster.labels_s
#质心
#centroid = cluster.cluster_centers_
#centroid.shape
fig = plt.figure()
ax = Axes3D(fig)
for i in range(9):
ax.scatter3D(data_new[y_pred == i,0],data_new[y_pred == i,1],data_new[y_pred == i,2],marker = ".")
ax.view_init(elev = 60,azim = 30)
ax.set_zlabel('Z')
ax.set_ylabel('Y')
ax.set_xlabel('X')
plt.show()
if __name__ == "__main__":
main()
说明,不能直接对图片进行处理,不支持超过二维的矩阵数组操作,先进行维度转换。