图聚类算法 matplotlib绘图 python程序

以下是一个简单的Python程序，使用matplotlib库和KMeans算法对图像进行聚类：

import matplotlib.pyplot as plt
import numpy as np
from sklearn.cluster import KMeans

# 读取图像数据
img = plt.imread('image.jpg')

# 将图像数据转换为二维数组
rows, cols, channels = img.shape
X = img.reshape(rows*cols, channels)

# 使用KMeans算法进行聚类
kmeans = KMeans(n_clusters=5, random_state=0).fit(X)
labels = kmeans.predict(X)
centers = kmeans.cluster_centers_

# 将聚类结果重新转换为图像数据
new_img = np.zeros_like(X)
for i in range(len(X)):
    new_img[i] = centers[labels[i]]
new_img = new_img.reshape(rows, cols, channels)

# 绘制原始图像和聚类结果图像
fig, ax = plt.subplots(1, 2, figsize=(10, 5))
ax[0].imshow(img)
ax[0].set_title('Original Image')
ax[1].imshow(new_img)
ax[1].set_title('Clustered Image')
plt.show()

该程序读取名为“image.jpg”的图像文件，将其转换为二维数组，然后使用KMeans算法将其聚类为5个簇。最后，程序将聚类结果重新转换为图像数据，并使用matplotlib库绘制原始图像和聚类结果图像。