如何使用python保存来自Kmeans输出的特定彩色图像

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

image = Image.open("image.jpg")

image_array = np.array(image)

reshaped_array = image_array.reshape(-1, 3)

kmeans = KMeans(n_clusters=K)
kmeans.fit(reshaped_array)

labels = kmeans.labels_

new_colors = kmeans.cluster_centers_[labels]

new_image_array = new_colors.reshape(image_array.shape)

new_image = Image.fromarray(new_image_array.astype('uint8'))
new_image.save("output_image.jpg")