python代码编写K-means聚类算法函数实现图像分割

好的，这个问题可以回答。以下是一个Python代码编写的K-means聚类算法函数实现图像分割的示例：

import numpy as np
from sklearn.cluster import KMeans

def image_segmentation(image, k):
    h, w, d = image.shape
    
    # Reshape the image to a 2D array of pixels
    pixels = image.reshape((h*w, d))
    
    # Run K-means clustering on the pixels
    kmeans = KMeans(n_clusters=k, random_state=0).fit(pixels)
    
    # Get the labels and reshape them back into an image
    labels = kmeans.labels_.reshape((h, w))
    
    # Convert the labels to a RGB image
    colors = np.array(kmeans.cluster_centers_, dtype=np.uint8)
    segmented_image = colors[labels]
    
    return segmented_image

这个函数使用了scikit-learn库中的KMeans类来执行K-means聚类算法，然后将聚类结果转换为图像格式返回。