W3School TIY Editor

# 这三行代码使编译器能够绘图：

import sys

import matplotlib

matplotlib.use('Agg')

​

import matplotlib.pyplot as plt

from sklearn.cluster import KMeans

​

# 定义样本数据：

x = [4, 5, 10, 4, 3, 11, 14, 6, 10, 12]  # 特征 1

y = [21, 19, 24, 17, 16, 25, 24, 22, 21, 21]  # 特征 2

​

# 组合特征数据：

data = list(zip(x, y))

​

# 创建 K-means 聚类模型（2 个簇）：

kmeans = KMeans(n_clusters=2)

​

# 训练模型：

kmeans.fit(data)

​

# 绘制聚类结果散点图（按簇标签着色）：

plt.scatter(x, y, c=kmeans.labels_)

​

# 显示图形：

plt.show()

​

# 这两行代码使编译器能够输出图形：

plt.savefig(sys.stdout.buffer)

sys.stdout.flush()