KMeans 聚类算法示例：将餐厅客户分类为 VIP 和普通客户

使用 KMeans 算法将餐厅客户分类为 VIP 和普通客户

问题描述： 假设某餐厅现有 4 个客户，分别编号为 a、b、c、d，这 4 位客户某段时间的平均消费水平分别为 1、2、4、5 万元。请用 KMeans 算法将这 4 个客户聚类为 VIP 客户和普通客户两个类别。

输入形式： 无

输出形式：

[0 0 1 1]

[[1.5]

 [4.5]]

样例输入：

无

样例输出：

[0 0 1 1]

[[1.5]

 [4.5]]

样例说明：

将 a、b 归为一类，c、d 归为一类，分别计算两类的平均消费水平得到 [1.5, 4.5]，输出聚类结果和两类的平均值。