使用Kmeans算法完成聚类任务

taoli-qiao2023-07-29 11:44

聚类任务

聚类任务是一种无监督学习任务，其目的是将一组数据点划分成若干个类别或簇，使得同一个簇内的数据点之间的相似度尽可能高，而不同簇之间的相似度尽可能低。聚类算法可以帮助我们发现数据中的内在结构和模式，发现异常点和离群值，简化数据表示，以及为进一步的分析提供基础。聚类任务在现实世界中有很多应用场景，以下是其中的一些例子：

  1. 市场细分：聚类可以帮助将市场分成不同的细分市场，以便更好地针对消费者需求制定营销策略。

  2. 图像分析：聚类可以用于图像分析，例如将相似的图像分组。

  3. 模式识别：聚类可以用于发现数据中的模式和关系，例如在医疗领域中，可以使用聚类来发现疾病之间的关系。

  4. 推荐系统：聚类可以用于推荐系统中，以将用户分组并向他们推荐相似的产品或服务。

K-Means算法

K-Means是一种基于聚类的无监督机器学习算法，其目的是将一组数据点分为k个不同的簇，使得每个数据点与其所属簇的中心点（也称质心）的距离最小化。以下是K-Means的工作原理：

  1. 初始化：随机选择k个数据点作为初始质心。

  2. 分配：对每个数据点,计算其与每个质心的距离，并将其分配给距离最近的质心所代表的簇。

  3. 重新计算质心：对于每个簇，重新计算其质心位置，即将该簇中所有数据点的坐标求平均。

  4. 重复执行第2,3步，直到所有数据点的簇分配不再改变或达到预设的最大迭代次数为止。

下面是用K-Means算法完成聚类的简单Demo，下面的demo中K设置为2.

python 复制代码 
from sklearn.cluster import KMeans
import numpy as np
# create some sample data
X = np.array([[1, 2], [1, 4], [1, 0], [4, 2], [4, 4], [4, 0]])
# create a KMeans object with 2 clusters
kmeans = KMeans(n_clusters=2, random_state=0)
# fit the KMeans object to the data
kmeans.fit(X)
# print the centroids of the two clusters
print(kmeans.cluster_centers_)
# predict the cluster labels for the data points
labels = kmeans.predict(X)
# print the predicted cluster labels for the data points
print(labels)

执行结果：cluster_centers_:[[1. 2.][4. 2.]], labels:[0 0 0 1 1 1]

上面的Demo中使用到KMeans函数，KMeans函数是一种聚类分析算法，用于将数据集分成多个簇。其主要作用是将相似的数据点分到同一个簇中，同时将不同的数据点分到不同的簇中。KMeans算法通过迭代寻找最优的聚类结果，可以对数据进行分组、分类和聚类分析。该函数包含多个输入参数，各个参数含义如下：

  • n_clusters：聚类的数量（簇的个数），即K值。默认值为8。如果知道数据的实际类别数目，可以将其设置为该数目；否则，可以通过手动设置不同的聚类数量来寻找最佳解。

  • **init：**初始化质心的方法。默认为"k-means++"，表示使用一种改进的贪心算法来选取初始质心。也可以设置为随机选择初始质心的"random"方法。

  • **max_iter：**最大迭代次数。默认值为300。当质心移动的距离小于阈值或达到最大迭代次数时，算法停止。

  • **tol：**质心移动的阈值。默认值为1e-4。当质心移动的距离小于该阈值时，算法停止。

  • **n_init：**随机初始化的次数。默认值为10。由于KMeans算法易受初始质心的影响，因此可以通过多次运行算法并选择最好的结果来减少随机性的影响。

  • **algorithm：**KMeans算法实现的方式。默认为"auto"，表示由算法自动选择最佳的实现方式（"full"表示使用标准的KMeans算法，"elkan"表示使用改进的Elkan算法）。对于大规模数据集，建议使用"elkan"实现方式。

上面的Demo例子是对List数据进行聚类，接下来看看如何使用K-means方法对足球队进行聚类，下面的例子中读取了csv文件中的原始数据，csv文件中存放了不同球队在三次比赛中的排名。

python 复制代码 
# coding: utf-8
from sklearn.cluster import KMeans
from sklearn import preprocessing
import pandas as pd
import numpy as np
# 输入数据
data = pd.read_csv('./kmeans/data.csv', encoding='gbk')
train_x = data[["2019年国际排名", "2018世界杯", "2015亚洲杯"]]
kmeans = KMeans(n_clusters=3)
# 规范化到[0,1]空间
min_max_scaler = preprocessing.MinMaxScaler()
train_x = min_max_scaler.fit_transform(train_x)
# kmeans算法
kmeans.fit(train_x)
predict_y = kmeans.predict(train_x)
# 合并聚类结果，插入到原数据中
result = pd.concat((data, pd.DataFrame(predict_y)), axis=1)
result.rename({0: u'聚类'}, axis=1, inplace=True)
print(result)

采用K-means方法进行聚类，假设K=3，聚类后的结果如下所示，可以看到把球队分到了0，1，2三种不同类型中。

对图像进行聚类

上面的例子是对数据进行聚类，下面看看如何对图像进行聚类，下面的Demo例子中将weixin登陆的图标按不同像素下的颜色分成了2类。

python 复制代码 
# -*- coding: utf-8 -*-
# 使用K-means对图像进行聚类，显示分割标识的可视化
import numpy as np
import PIL.Image as image
from sklearn.cluster import KMeans
from sklearn import preprocessing


# 加载图像，并对数据进行规范化
def load_data(filePath):
    # 读文件
    f = open(filePath, 'rb')
    data = []
    # 得到图像的像素值
    img = image.open(f)
    # 得到图像尺寸
    width, height = img.size
    for x in range(width):
        for y in range(height):
            # 得到点(x,y)的三个通道值
            c1, c2, c3 = img.getpixel((x, y))
            data.append([c1, c2, c3])
    f.close()
    # 采用Min-Max规范化
    mm = preprocessing.MinMaxScaler()
    data = mm.fit_transform(data)
    return np.asarray(data), width, height

# 加载图像，得到规范化的结果img，以及图像尺寸
img, width, height = load_data('./kmeans/weixin.jpg')

# 用K-Means对图像进行2聚类
kmeans = KMeans(n_clusters=3)
kmeans.fit(img)
label = kmeans.predict(img)
# 将图像聚类结果，转化成图像尺寸的矩阵
label = label.reshape([width, height])
# 创建个新图像pic_mark，用来保存图像聚类的结果，并设置不同的灰度值
pic_mark = image.new("L", (width, height))
for x in range(width):
    for y in range(height):
        # 根据类别设置图像灰度, 类别0 灰度值为255， 类别1 灰度值为127
        pic_mark.putpixel((x, y), int(256 / (label[x][y] + 1)) - 1)
pic_mark.save("./kmeans/weixin_mark1.jpg", "JPEG")

下图中第一张图是原图，第二张图是分类K=2的结果。可以看到，因为只进行了2种类型区分，新生成的图片中，纯白色是原图中深蓝色的代表，黑灰色是原图中白亮色的代表。说明聚类正确。

图三是K=16的分类结果，当分类K=16时，和原图就很接近了，K=16的分类代码细节如下所示：

python 复制代码 
# -*- coding: utf-8 -*-
# 使用K-means对图像进行聚类，并显示聚类压缩后的图像
import numpy as np
import PIL.Image as image
from sklearn.cluster import KMeans
from sklearn import preprocessing
import matplotlib.image as mpimg


# 加载图像，并对数据进行规范化
def load_data(filePath):
    # 读文件
    f = open(filePath, 'rb')
    data = []
    # 得到图像的像素值
    img = image.open(f)
    # 得到图像尺寸
    width, height = img.size
    for x in range(width):
        for y in range(height):
            # 得到点(x,y)的三个通道值
            c1, c2, c3 = img.getpixel((x, y))
            data.append([(c1 + 1) / 256.0, (c2 + 1) / 256.0, (c3 + 1) / 256.0])
    f.close()
    return np.asarray(data), width, height


# 加载图像，得到规范化的结果imgData，以及图像尺寸
img, width, height = load_data('./kmeans/weixin.jpg')
# 用K-Means对图像进行16聚类
kmeans = KMeans(n_clusters=16)
label = kmeans.fit_predict(img)
# 将图像聚类结果，转化成图像尺寸的矩阵
label = label.reshape([width, height])
# 创建个新图像img，用来保存图像聚类压缩后的结果
img = image.new('RGB', (width, height))
for x in range(width):
    for y in range(height):
        c1 = kmeans.cluster_centers_[label[x, y], 0]
        c2 = kmeans.cluster_centers_[label[x, y], 1]
        c3 = kmeans.cluster_centers_[label[x, y], 2]
        img.putpixel((x, y),
                     (int(c1 * 256) - 1, int(c2 * 256) - 1, int(c3 * 256) - 1))
img.save('./kmeans/weixin_new.jpg')

上面介绍了如何使用K-Means算法完成文本类或者图片类聚类任务，在实际项目中，K-Means算法应用非常广泛，主要应用在如下的业务场景中。

  1. 市场营销：K-Means算法可以对市场消费者进行分类，以便公司更好地了解他们的需求和行为，制定更有效的营销策略。

  2. 图像处理：K-Means算法可以用于对图像像素进行聚类，以实现图像压缩和图像分割等功能。

  3. 自然语言处理：K-Means算法可以用于对文本数据进行聚类，以实现语义分析和文本分类等功能。

  4. 生物信息学：K-Means算法可以用于对生物数据进行聚类，以实现基因分类和蛋白质分类等功能。

  5. 金融领域：K-Means算法可以用于对金融数据进行聚类，以实现风险评估和资产管理等功能。

上面提到K-Means算法可以对图像像素进行聚类，以实现图像压缩的功能，下面的例子中就采用K-Means算法对图片像素进行聚类，从而实现压缩的效果。

python 复制代码 
import numpy as np
import matplotlib.pyplot as plt
from sklearn.cluster import KMeans
from PIL import Image

# 加载图片
img = Image.open('./kmeans/baby.jpg')
img_data = np.array(img)

# 将三维的图片数组变成二维的像素点数组
pixels = img_data.reshape(
    (img_data.shape[0] * img_data.shape[1], img_data.shape[2]))
# 使用K-Means聚类算法对像素点进行聚类
kmeans = KMeans(n_clusters=16, random_state=0)
labels = kmeans.fit_predict(pixels)

# 将每个像素点替换为所属聚类的中心点
new_pixels = kmeans.cluster_centers_[labels]

# 将一维的像素点数组还原为图片数组的形式
new_img_data = new_pixels.reshape(
    (img_data.shape[0], img_data.shape[1], img_data.shape[2]))

# 显示原始图片和压缩后的图片
fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(12, 6))
fig.suptitle('Image Compression using K-Means Clustering')

ax1.set_title('Original Image')
ax1.imshow(img_data)

ax2.set_title('Compressed Image')
ax2.imshow(new_img_data.astype('uint8'))

plt.show()

原图和压缩后的图片结果如下所示：

相关推荐
1 9 J27 分钟前
Java 上机实践4（类与对象）
java·开发语言·算法
passer__jw7672 小时前
【LeetCode】【算法】3. 无重复字符的最长子串
算法·leetcode
passer__jw7672 小时前
【LeetCode】【算法】21. 合并两个有序链表
算法·leetcode·链表
sweetheart7-72 小时前
LeetCode22. 括号生成（2024冬季每日一题 2）
算法·深度优先·力扣·dfs·左右括号匹配
封步宇AIGC5 小时前
量化交易系统开发-实时行情自动化交易-3.4.2.Okex行情交易数据
人工智能·python·机器学习·数据挖掘
封步宇AIGC5 小时前
量化交易系统开发-实时行情自动化交易-2.技术栈
人工智能·python·机器学习·数据挖掘
景鹤5 小时前
【算法】递归+回溯+剪枝：78.子集
算法·机器学习·剪枝
_OLi_5 小时前
力扣 LeetCode 704. 二分查找（Day1：数组）
算法·leetcode·职场和发展
丶Darling.5 小时前
Day40 | 动态规划 ：完全背包应用 组合总和IV（类比爬楼梯）
c++·算法·动态规划·记忆化搜索·回溯
风影小子5 小时前
IO作业5
算法
热门推荐
01聊聊 Python 中的同步原语，为什么有了 GIL 还需要同步原语02【HarmonyOS】HUAWEI DevEco Studio 下载地址汇总03Windows10安装PCL1.14.0及点云配准04C++——超强级病毒代码05全面解析：构建基于深度学习的安全帽检测系统（UI界面+YOLO代码+数据集）06组基轨迹建模 GBTM的介绍与实现（Stata 或 R）07硬件预取之空间预取08Ubuntu 20.04使用Livox mid 360 测试 FAST_LIO09Coze扣子平台完整体验和实践（附国内和国际版对比）10【SCS-CN】SCS-CN模型中CN值的确定