【人工智能】—基于K-Means算法商场顾客聚类实战教程

在这篇博文之前一直是给大家做机器学习有监督学习教程,今天来一篇无监督学习教程。

K-Means算法是一种基于中心的聚类方法,它试图找到数据点的K个簇,使得簇内的数据点尽可能相似,而簇间的数据点尽可能不同。下面是K-Means算法的详细解释和步骤:

算法原理:
K-Means算法基于最小化簇内误差平方和(Within-Cluster Sum of Squares, WCSS)来划分簇。簇内误差平方和是簇内所有点到簇中心的距离平方的总和。算法的目标是最小化所有簇的WCSS。

算法步骤:

  1. 初始化:选择K个数据点作为初始的簇中心。这些点可以是随机选择的,也可以通过更智能的方法选择,如K-Means++算法。
  2. 分配:将每个数据点分配到最近的簇中心,形成K个簇。这一步中,每个数据点根据其与簇中心的距离被分配到最近的簇。
  3. 更新簇中心:重新计算每个簇的中心点。簇中心是该簇内所有数据点的均值(或中位数,取决于距离度量方式)。
  4. 迭代:重复步骤2和3,直到满足以下任一条件:
    (1)簇中心的位置变化非常小,即算法收敛。
    (2)达到预设的最大迭代次数。
  5. 结束:当簇中心不再显著变化或达到最大迭代次数时,算法结束,此时的簇划分被认为是最终结果。

算法细节:

  • 距离度量:K-Means通常使用欧几里得距离来衡量数据点之间的相似性,但也可以根据不同的应用场景使用曼哈顿距离、闵可夫斯基距离等。
  • 选择K值:K值的选择通常依赖于领域知识或通过肘部法则(Elbow
    Method)等方法来确定。肘部法则通过观察不同K值对应的WCSS,选择WCSS下降速率骤减的点作为K值。
  • 初始簇中心选择:初始簇中心的选择对算法的收敛性和最终结果有很大影响。K-Means++是一种改进的初始化方法,它通过概率方式选择初始簇中心,以减少陷入局部最优解的风险。
  • 算法变体:K-Means算法有多种变体,如K-Medoids算法,它使用实际数据点作为簇中心,而不是计算出的均值,这在处理离群点时更为稳健。

应用场景:

K-Means算法在多个领域都有应用,包括但不限于:

  • 图像处理:用于图像分割,将图像划分为不同的区域。
  • 文本分析:用于文本聚类,将相似的文档分到同一个簇。
  • 生物信息学:用于基因表达数据的分析,识别具有相似表达模式的基因。
  • 市场分析:用于客户细分,识别具有相似购买行为的客户群体。

局限性:

  • 对噪声敏感:K-Means对异常值和噪声较为敏感,可能会影响簇中心的计算和最终的聚类结果。
  • 对初始条件敏感:算法可能会收敛到局部最优解,而不是全局最优解。
  • 不适合非球形簇:K-Means假设簇是凸形的,对于非球形或有复杂形状的簇,聚类效果可能不佳。

数据集情况

通过超市商场会员卡,获取客户的基本数据,该数据集仅用于学习客户细分概念,也称为市场篮子分析来源kaggle(数据已云游至上方王母娘娘蟠桃园,免费摘取)。

数据集概述:

  • 来源:Kaggle平台
  • 目的:用于商场顾客细分的数据分析,旨在帮助商场更深入地了解顾客群体,以制定更精准的营销策略和服务。
  • 数据规模:包含200条数据记录,共5个字段(变量)。

数据字段:

  • 客户ID(CustomerID):唯一标识每个客户的编号。
  • 性别(Gender):客户的性别,分为’Male’和’Female’两类。
  • 年龄(Age):客户的年龄,以整数形式表示。
  • 年收入(AnnualIncome):客户的年收入,以千美元为单位(k$)。
  • 支出分数(SpendingScore):商场根据客户的消费行为和购买数据分配的分数,范围为1-100。

应用场景:

  • 顾客细分:通过数据分析,可以将顾客细分为不同的群体,如高收入高消费群体、年轻消费群体等。
  • 营销策略制定:基于顾客细分的结果,商场可以制定有针对性的营销策略,如针对高收入群体推出高端商品或服务,针对年轻群体推出时尚或潮流商品。
  • 用户画像构建:通过对数据的深入挖掘,可以构建出详细的用户画像,帮助商场更全面地了解顾客的需求和偏好。

初步分析:

导入数据:

import numpy as np 
import pandas as pd 
import matplotlib.pyplot as plt 
import seaborn as sns 
import plotly as py
import plotly.graph_objs as go
from sklearn.cluster import KMeans
import warnings
import os
warnings.filterwarnings("ignore")
plt.rcParams ['font.sans-serif'] ='SimHei'      #显示中文
plt.rcParams ['axes.unicode_minus']=False       #显示负号
df = pd.read_csv('Mall_Customers.csv')
df.head()

在这里插入图片描述描述性统计:
在这里插入图片描述在这里插入图片描述描述性统计的意思就是我不想打字,自己看了:)

# 使用'fivethirtyeight'样式来设置绘图的主题和样式,这个样式常用于《五三八》(FiveThirtyEight)网站的图表风格  
plt.style.use('fivethirtyeight')  
# 创建一个图形窗口,并设置其编号(虽然在这里编号并不关键,因为只有一个窗口)和大小(宽度为15英寸,高度为6英寸)  
plt.figure(1, figsize=(15, 6))    
# 初始化一个计数器,用于后面的subplot布局  
n = 0    
# 对数据集中的'Age'、'Annual Income (k$)'和'Spending Score (1-100)'这三个字段进行迭代  
for x in ['Age', 'Annual Income (k$)', 'Spending Score (1-100)']:  # 每次迭代,计数器n加1  n += 1        # 创建一个子图(subplot),在1行3列的布局中,这是第n个子图  # 第一个参数是子图的行数,第二个参数是列数,第三个参数是当前子图的编号(从1开始)  plt.subplot(1, 3, n)        # 调整子图之间的水平和垂直间距,使得子图之间不会过于拥挤或过于疏远  plt.subplots_adjust(hspace=0.5, wspace=0.5)        # 使用seaborn库的distplot函数绘制df[x]列的分布图,bins参数指定了分布的条形数量  # 这里假设df是一个pandas DataFrame,且包含了'Age'、'Annual Income (k$)'和'Spending Score (1-100)'这三个字段  sns.distplot(df[x], bins=20)        # 设置当前子图的标题,格式为"Distplot of 字段名"  plt.title('Distplot of {}'.format(x))    
# 显示所有子图  
plt.show()

绘制年龄、年收入、支出分数直方图:
在这里插入图片描述

# 创建一个新的图形窗口,编号为1,并设置其大小为宽度15英寸,高度5英寸  
plt.figure(1, figsize=(15, 5))    
# 使用seaborn库中的countplot函数绘制一个条形图  
# 其中,y参数指定了用于分类的列名(在这里是'Gender'),即条形图的分类标签  
# data参数指定了数据源,即包含数据的pandas DataFrame(在这里是df)  
sns.countplot(y='Gender', data=df)    
# 显示图形窗口中的所有图形  
plt.show()

绘制性别计数图:
在这里插入图片描述

# 创建一个新的图形窗口,编号为1,并设置其大小为宽度15英寸,高度7英寸  
plt.figure(1, figsize=(15, 7))    
# 初始化一个计数器,用于后面的subplot布局  
n = 0    
# 遍历'Age'、'Annual Income (k$)'和'Spending Score (1-100)'这三个字段,作为x轴和y轴的可能选择  
for x in ['Age' , 'Annual Income (k$)' , 'Spending Score (1-100)']:  for y in ['Age' , 'Annual Income (k$)' , 'Spending Score (1-100)']:  # 每次循环,计数器n加1  n += 1            # 创建一个子图(subplot),在3行3列的布局中,这是第n个子图  # 注意:这里会生成9个子图(3x3),但其中很多可能是重复的或没有意义的(例如x和y都是'Age')  plt.subplot(3, 3, n)            # 调整子图之间的水平和垂直间距  plt.subplots_adjust(hspace=0.5, wspace=0.5)            # 使用seaborn的regplot函数绘制x和y之间的回归线及其置信区间  # data参数指定了数据源,即包含数据的pandas DataFrame(在这里是df)  sns.regplot(x=x, y=y, data=df)            # 设置y轴的标签  # 如果y是一个包含空格的字符串(例如'Annual Income (k$)'),则使用split()方法将其分割为两部分,并重新组合以作为标签  # 如果y只包含一个词(例如'Age'),则直接使用y作为标签  plt.ylabel(y.split()[0]+' '+y.split()[1] if len(y.split()) > 1 else y)    
# 显示所有子图  
plt.show()

绘制年龄、年收入和支出得分之间的关系图:
在这里插入图片描述

# 创建一个新的图形窗口,编号为1,并设置其大小为宽度15英寸,高度6英寸  
plt.figure(1, figsize=(15, 6))    
# 遍历'Male'和'Female'两个性别  
for gender in ['Male', 'Female']:  # 使用scatter函数绘制散点图  # x参数指定了x轴的数据列名,这里是'Age'  # y参数指定了y轴的数据列名,这里是'Annual Income (k$)'  # data参数指定了数据源,这里是从df中筛选出'Gender'列等于当前循环中gender的数据子集  # s参数指定了散点的大小,这里是200  # alpha参数指定了散点的透明度,这里是0.5,意味着散点会有一定的重叠效果  # label参数为散点设置了标签,用于在图例中显示  plt.scatter(x='Age', y='Annual Income (k$)', data=df[df['Gender'] == gender],  s=200, alpha=0.5, label=gender)    
# 设置x轴的标签为'Age'  
plt.xlabel('Age')    
# 设置y轴的标签为'Annual Income (k$)'  
plt.ylabel('Annual Income (k$)')    
# 设置图形的标题为'Age vs Annual Income w.r.t Gender'  
plt.title('Age vs Annual Income w.r.t Gender')    
# 显示图例,图例中的标签来源于前面scatter函数中设置的label参数  
plt.legend()    
# 显示图形窗口中的所有图形  
plt.show()

绘制年龄与性别(男和女)的散点关系图:
在这里插入图片描述

# 创建一个新的图形窗口,编号为1,并设置其大小为宽度15英寸,高度6英寸  
plt.figure(1, figsize=(15, 6))    
# 遍历'Male'和'Female'两个性别  
for gender in ['Male', 'Female']:  # 使用scatter函数绘制散点图  # x参数指定了x轴的数据列名,这里是'Annual Income (k$)'  # y参数指定了y轴的数据列名,这里是'Spending Score (1-100)'  # data参数指定了数据源,这里是从df中筛选出'Gender'列等于当前循环中gender的数据子集  # s参数指定了散点的大小,这里是200  # alpha参数指定了散点的透明度,这里是0.5,意味着散点会有一定的重叠效果  # label参数为散点设置了标签,用于在图例中显示  plt.scatter(x='Annual Income (k$)', y='Spending Score (1-100)',  data=df[df['Gender'] == gender], s=200, alpha=0.5, label=gender)    
# 设置x轴的标签为'Annual Income (k$)'  
plt.xlabel('Annual Income (k$)')    
# 注意:这里使用了逗号来将两个函数调用分隔开,但这不是必要的。可以单独调用plt.ylabel  
# 设置y轴的标签为'Spending Score (1-100)'  
plt.ylabel('Spending Score (1-100)')    
# 设置图形的标题为'Annual Income vs Spending Score w.r.t Gender'  
plt.title('Annual Income vs Spending Score w.r.t Gender')    
# 显示图例,图例中的标签来源于前面scatter函数中设置的label参数  
plt.legend()    
# 显示图形窗口中的所有图形  
plt.show()

绘制年收入、支出分数的性别(男和女)散点关系图:
在这里插入图片描述

# 创建一个新的图形窗口,编号为1,并设置其大小为15x7英寸  
plt.figure(1 , figsize = (15 , 7))    
# 初始化一个计数器n,用于跟踪子图的编号  
n = 0     
# 遍历列名列表,这些列是我们希望根据性别来绘制小提琴图和蜂群图的特征  
for cols in ['Age' , 'Annual Income (k$)' , 'Spending Score (1-100)']:  # 每次循环,n增加1,以便在子图中定位  n += 1         # 使用subplot创建一个子图,1行3列,当前是第n个子图  plt.subplot(1 , 3 , n)        # 调整子图之间的水平和垂直间距,使图形之间有足够的空间  plt.subplots_adjust(hspace = 0.5 , wspace = 0.5)        # 使用seaborn的violinplot函数绘制小提琴图,小提琴图展示了数据的分布  # x轴是列名cols对应的值,y轴是'Gender'列的值,数据来自df  # palette参数设置了小提琴图的颜色方案  sns.violinplot(x = cols , y = 'Gender' , data = df , palette = 'vlag')        # 使用seaborn的swarmplot函数绘制蜂群图,蜂群图展示了每个数据点的位置  # x轴和y轴与小提琴图相同,数据也来自df  sns.swarmplot(x = cols , y = 'Gender' , data = df)        # 如果这是第一个子图,则设置y轴的标签为'Gender',否则不设置  plt.ylabel('Gender' if n == 1 else '')        # 如果这是第二个子图,则设置标题为'Boxplots & Swarmplots',否则不设置  # 注意:虽然标题中提到了Boxplots,但实际上并没有绘制箱线图  plt.title('Boxplots & Swarmplots' if n == 2 else '')    
# 显示图形  
plt.show()

绘制年龄、年收入、支出分数的性别(男和女)小提琴图:
在这里插入图片描述
**小提琴图(Violin Plot)**是一种用于展示数据分布的图形,它结合了箱线图(Box Plot)和密度图(Density Plot)的特点。小提琴图提供了数据的多种统计信息,包括数据的分布、密度、中位数、四分位数以及异常值等。以下是小提琴图的一些关键特点和组成部分:

  • 形状:小提琴图的主体部分是曲线,形状像小提琴的侧面,因此得名。曲线的宽度表示数据点的密度,即在该范围内数据点的数量。
  • 箱线图部分:在小提琴图的中心,通常会有一个箱线图,显示数据的中位数、四分位数和异常值。这与标准的箱线图相同。
  • 数据分布:小提琴图的曲线部分显示了数据的密度分布。曲线越宽,表示在该区间内数据点越密集。
  • 颜色:小提琴图可以使用颜色来表示数据的密度,颜色越深表示密度越高。
  • 多组数据对比:小提琴图非常适合用于展示和比较多组数据的分布情况。例如,可以在同一张图上展示不同条件下的数据分布,通过比较小提琴图的形状和大小来直观地看出不同组数据的差异。
  • 适用场景:小提琴图适用于展示连续数据的分布情况,尤其是当数据集较大,需要同时展示多个维度的数据时。
  • 局限性:由于小提琴图包含了大量的信息,对于初学者来说可能不太容易解读。此外,当数据集中包含多个峰值或多个模式时,小提琴图可能不如其他图形(如直方图)直观。
# 注释:从df中选取'Age'和'Spending Score (1-100)'两列,  
# 并将其转换为numpy数组(因为KMeans需要数值型数组作为输入)  
# .iloc[: , :] 是一个冗余的索引,因为这里它选择了所有的行和列(但在这个上下文中,它仍然是有效的)  
X1 = df[['Age' , 'Spending Score (1-100)']].iloc[: , :].values    
# 注释:初始化一个空列表,用于存储不同聚类数量下KMeans算法的惯性(inertia)值  
# 惯性是KMeans算法的一个指标,表示所有样本到其最近质心的距离的平方和  
inertia = []    
# 注释:开始一个循环,从1到10(包含1但不包含11),用于测试不同的聚类数量  
for n in range(1 , 11):  # 注释:在每次循环中,创建一个KMeans对象,设置n_clusters为当前循环的n值  # init='k-means++' 表示使用k-means++初始化质心  # n_init=10 表示算法将运行10次,并返回最佳结果(基于惯性)  # max_iter=300 表示算法的最大迭代次数  # tol=0.0001 是收敛的容忍度,即当质心变化小于这个值时,算法将停止  # random_state=111 确保了每次运行的结果都是可重复的  # algorithm='elkan' 是一个加速KMeans算法的选项  algorithm = (KMeans(n_clusters = n ,init='k-means++', n_init = 10 ,max_iter=300,   tol=0.0001,  random_state= 111  , algorithm='elkan') )        # 注释:使用当前KMeans对象对X1数据进行拟合,即进行聚类  algorithm.fit(X1)        # 注释:将KMeans算法拟合后的惯性值添加到inertia列表中  # 惯性值越小,通常表示聚类效果越好(但需要注意过拟合的可能性)  inertia.append(algorithm.inertia_)

这段代码的目的是通过计算不同聚类数量下的KMeans惯性值,来评估并找出最佳的聚类数量,从而优化KMeans聚类的效果。通常,惯性值越小表示聚类效果越好,但需避免过拟合。

# 创建一个新的图形窗口,编号为1,并设置其大小为15x6英寸  
plt.figure(1, figsize=(15, 6))    
# 绘制点图,横坐标是聚类数量的范围(从1到10),纵坐标是对应的惯性值  
# 'o' 表示数据点将被绘制为圆圈  
plt.plot(np.arange(1, 11), inertia, 'o')    
# 绘制线图,与点图使用相同的横纵坐标数据  
# '-' 表示线将被绘制为实线  
# alpha=0.5 设置线的透明度为0.5,使其半透明,以便与点图区分  
plt.plot(np.arange(1, 11), inertia, '-', alpha=0.5)    
# 设置x轴的标签为'Number of Clusters',表示聚类数量  
# 设置y轴的标签为'Inertia',表示惯性值  
# plt.xlabel() 和 plt.ylabel() 函数返回None,但在这里我们使用逗号将它们组合在一行,这是一种常见的Python编程风格,用于在一行中执行多个操作  
plt.xlabel('Number of Clusters'), plt.ylabel('Inertia')    
# 显示图形  
plt.show()

这段代码可视化KMeans聚类算法中不同聚类数量对应的惯性值,展示了KMeans聚类中聚类数量与惯性值之间的关系,帮助用户确定最佳的聚类数量。

KMeans聚类中聚类数量与惯性值之间的关系图:
在这里插入图片描述

# 创建一个KMeans算法对象  
# 参数说明:  
#   n_clusters=4: 指定要形成的簇的数量为4  
#   init='k-means++': 使用k-means++算法来初始化质心  
#   n_init=10: 算法将使用不同的质心初始化运行10次,并选择最佳结果(基于惯性)  
#   max_iter=300: 算法的最大迭代次数  
#   tol=0.0001: 收敛的容忍度,即当质心变化小于这个值时,算法将停止  
#   random_state=111: 随机数生成器的种子,确保结果的可重复性  
#   algorithm='elkan': 使用Elkan的K-Means算法,它使用三角形不等式来加速计算  
algorithm = (KMeans(n_clusters=4, init='k-means++', n_init=10, max_iter=300,   tol=0.0001, random_state=111, algorithm='elkan'))    
# 使用KMeans算法对象对X1数据进行拟合,即进行聚类  
# X1是一个包含'Age'和'Spending Score (1-100)'的二维numpy数组  
algorithm.fit(X1)    
# 获取每个数据点的聚类标签  
# labels1是一个与X1数据点数量相同的数组,其中每个元素表示对应数据点所属的簇的编号(从0到n_clusters-1)  
labels1 = algorithm.labels_    
# 获取每个簇的质心  
# centroids1是一个形状为(n_clusters, n_features)的二维数组  
# 在这个例子中,由于n_clusters=4且n_features=2('Age'和'Spending Score'两列),所以centroids1的形状为(4, 2)  
# 每一行代表一个簇的质心,包含两个值:该簇的平均'Age'和平均'Spending Score'  
centroids1 = algorithm.cluster_centers_

以上代码创建了一个KMeans聚类算法对象,并设置了多个参数来指导聚类过程。其中,指定了簇的数量为4,并使用k-means++算法进行质心初始化,以增加聚类的稳定性和效率。算法将尝试10种不同的质心初始化,并选择其中惯性最小的结果作为最佳聚类。此外,还设置了最大迭代次数和收敛容忍度来控制算法的停止条件。使用指定的随机数种子确保了结果的可重复性。算法被应用于一个二维数据集X1,该数据集包含’Age’和’Spending Score’两个特征。聚类完成后,代码获取了每个数据点的聚类标签以及每个簇的质心。这些质心表示了每个簇中数据点的平均值,可以帮助理解各个簇的特征。

# 设置网格步长,这个值决定了网格的粒度或精细程度  
h = 0.02    
# 找到X1中第一列(即'Age'列)的最小值和最大值,并分别减去和加上1,以扩展网格的范围  
# 这样做是为了确保所有的数据点都包含在网格内,尤其是在绘制边界时  
x_min, x_max = X1[:, 0].min() - 1, X1[:, 0].max() + 1    
# 找到X1中第二列(即'Spending Score (1-100)'列)的最小值和最大值,并分别减去和加上1  
# 同样是为了扩展网格的范围  
y_min, y_max = X1[:, 1].min() - 1, X1[:, 1].max() + 1    
# 使用numpy的meshgrid函数在(x_min, x_max)和(y_min, y_max)之间创建一个二维网格  
# np.arange(x_min, x_max, h) 和 np.arange(y_min, y_max, h) 分别生成x和y轴上的值  
# meshgrid函数将这些一维数组转换为两个二维数组xx和yy,它们分别表示网格的x和y坐标  
xx, yy = np.meshgrid(np.arange(x_min, x_max, h), np.arange(y_min, y_max, h))    
# 将xx和yy的二维数组平铺(ravel)为一维数组,并将它们沿着列方向(使用np.c_)合并成一个二维数组  
# 这个二维数组的每一行代表网格上的一个点的坐标(x, y)  
# 这是为了将网格上的点作为输入数据传递给KMeans算法的predict方法  
Z = algorithm.predict(np.c_[xx.ravel(), yy.ravel()])    
# 经过predict方法处理后,Z是一个一维数组,包含了网格上每个点所属的簇的编号  
# 接下来,通常会将Z重新整形为与xx和yy相同的二维形状,以便在绘制图形时能够正确对应每个网格点所属的簇  
# 这通常是通过Z = Z.reshape(xx.shape)来实现的,但您提供的代码段中没有这一步

以上代码的主要目的是在包含’Age’和’Spending Score (1-100)'特征的数据集X1的范围内创建一个二维网格,并使用先前训练的KMeans算法(algorithm)来预测这个网格上每个点所属的簇。首先,通过扩展数据集中这两个特征的最小值和最大值来定义网格的边界。然后,使用numpy的meshgrid函数创建了一个二维网格,该网格覆盖了由x_min到x_max和y_min到y_max定义的区域,网格的粒度由步长h控制。接着,将网格的x和y坐标合并成一个二维数组,作为输入数据传递给KMeans算法的predict方法,以获取每个网格点所属的簇的编号。最后,通常会将预测的簇编号重新整形为与网格相同的二维形状,以便在可视化时能够正确对应每个网格点所属的簇。

# 创建一个新的图形窗口,编号为1,并设置其大小为15x7英寸  
plt.figure(1 , figsize = (15 , 7) )
# 清除当前图形窗口中的所有内容(如果存在的话),这确保我们从一个干净的画布开始  
plt.clf()
# 将之前预测得到的Z数组(包含网格上每个点所属的簇编号)重新整形为与xx(网格的x坐标)相同的二维形状  
Z = Z.reshape(xx.shape)
# 使用imshow函数绘制Z数组,即绘制出聚类结果  
# interpolation='nearest':表示使用最近邻插值法来绘制图像  
# extent=(xx.min(), xx.max(), yy.min(), yy.max()):定义图像的范围  
# cmap=plt.cm.Pastel2:选择颜色映射(这里选择了Pastel2颜色映射)  
# aspect='auto':自动调整x和y轴的缩放比例  
# origin='lower':设置坐标原点在图像的左下角  
plt.imshow(Z , interpolation='nearest', extent=(xx.min(), xx.max(), yy.min(), yy.max()),cmap = plt.cm.Pastel2, aspect = 'auto', origin='lower')
# 绘制原始数据点,这里使用scatter函数  
# x 和 y 参数通常应该是具体的数值,但这里使用了字符串'Age'和'Spending Score (1-100)'  
# 这意味着我们要从DataFrame df中引用这些列。但是,为了保持一致性,我们最好直接使用列名对应的数值数组  
# c=labels1:设置数据点的颜色,根据labels1中的簇编号进行着色  
# s=200:设置数据点的大小为200 
plt.scatter( x = 'Age' ,y = 'Spending Score (1-100)' , data = df , c = labels1 , s = 200 )
# 绘制聚类中心(质心)  
# x 和 y 分别是质心的x和y坐标  
# s=300:设置质心点的大小为300  
# c='red':设置质心点的颜色为红色  
# alpha=0.5:设置质心点的透明度为0.5  
plt.scatter(x = centroids1[: , 0] , y =  centroids1[: , 1] , s = 300 , c = 'red' , alpha = 0.5)
# 设置y轴的标签为'Spending Score (1-100)'  
# 设置x轴的标签为'Age'  
# 使用逗号操作符将两个设置标签的操作放在一行中,这是一种常见的Python编程风格  
plt.ylabel('Spending Score (1-100)') , plt.xlabel('Age')
plt.show()

年龄和支出分数的可视化KMeans聚类的结果:

在这里插入图片描述
后面会重复不同变量的聚类结果,由于时间有限,我就直接上代码和聚类的结果。

# 从df数据集中选择'Annual Income (k$)'和'Spending Score (1-100)'两列,并将其作为numpy数组赋值给X2  
# iloc[: , :]表示选择所有的行和所有的列(这里其实是冗余的,但明确表明了意图)  
X2 = df[['Annual Income (k$)' , 'Spending Score (1-100)']].iloc[: , :].values    
# 初始化一个空列表inertia,用于存储不同聚类数量下的惯性值  
inertia = []    
# 遍历从1到10的整数,作为KMeans算法中可能的聚类数量n  
for n in range(1 , 11):  # 创建一个KMeans对象,设置n_clusters为当前的n值,使用'k-means++'作为初始化方法  # 设置n_init为10,意味着会进行10次不同的初始化并选择最优结果  # max_iter设置为300,即算法的最大迭代次数  # tol设置为0.0001,即算法收敛的容忍度  # random_state设置为111,以确保结果的可复现性  # 使用'elkan'作为KMeans的加速算法  algorithm = (KMeans(n_clusters = n ,init='k-means++', n_init = 10 ,max_iter=300,   tol=0.0001,  random_state= 111  , algorithm='elkan') )        # 使用KMeans对象对X2进行拟合(即进行聚类)  algorithm.fit(X2)        # 将当前KMeans对象的惯性值(即聚类内部点到其质心的距离的平方和)添加到inertia列表中  inertia.append(algorithm.inertia_)
# 创建一个新的图形窗口,编号为1,并设置其大小为15x6英寸  
plt.figure(1, figsize=(15, 6))    
# 绘制散点图,其中x轴是np.arange(1, 11)(即从1到10的整数数组),y轴是之前计算得到的inertia(惯性值列表)  
# 'o'表示使用圆圈作为数据点的标记  
plt.plot(np.arange(1, 11), inertia, 'o')    
# 在同一图形上再次绘制线条图,使用相同的x和y数据  
# '-'表示使用实线连接数据点  
# alpha=0.5设置线条的透明度为0.5,使其半透明  
plt.plot(np.arange(1, 11), inertia, '-', alpha=0.5)    
# 设置x轴的标签为'Number of Clusters'(簇的数量)  
# 使用逗号操作符将两个操作放在同一行上,这是一种常见的Python编程风格  
plt.xlabel('Number of Clusters'), plt.ylabel('Inertia')    
# 显示图形  
plt.show()

KMeans聚类中聚类数量与惯性值之间的关系图:
在这里插入图片描述

# 创建一个KMeans对象,设置其参数:  
# n_clusters=5:表示要形成的簇的数量为5  
# init='k-means++':使用'k-means++'作为初始化质心的方法  
# n_init=10:在给定n_clusters的情况下,进行10次KMeans算法的运行,并选择最佳结果(即惯性最小的结果)  
# max_iter=300:KMeans算法的最大迭代次数为300  
# tol=0.0001:KMeans算法收敛的容忍度,即当质心变化小于这个值时,算法停止迭代  
# random_state=111:设置随机种子为111,以确保结果的可复现性  
# algorithm='elkan':使用'elkan'算法,这是KMeans算法的一种优化版本,尤其在稀疏数据上表现更好  
algorithm = (KMeans(n_clusters = 5 ,init='k-means++', n_init = 10 ,max_iter=300,   tol=0.0001,  random_state= 111  , algorithm='elkan') )    
# 使用前面定义的KMeans对象(algorithm)对X2(包含'Annual Income (k$)'和'Spending Score (1-100)'的数据)进行拟合,即进行聚类  
algorithm.fit(X2)    
# 从KMeans对象中获取每个数据点的簇标签,即每个数据点所属的簇的编号  
# 这些标签是整数,从0开始,对应于簇的索引  
labels2 = algorithm.labels_    
# 从KMeans对象中获取每个簇的质心(即中心点)的坐标  
# 这是一个二维数组,其中每一行代表一个簇的质心,列对应于数据集中的特征(这里是'Annual Income (k$)'和'Spending Score (1-100)')  
centroids2 = algorithm.cluster_centers_
# 设置网格步长,即网格中相邻两个点之间的间隔。这里设置为0.02,表示网格的精细度。  
h = 0.02    
# 找出数据集X2中第一列(即'Annual Income (k$)')的最小值和最大值,并分别减1和加1来扩展网格的边界,以确保所有数据点都被包含在内。  
x_min, x_max = X2[:, 0].min() - 1, X2[:, 0].max() + 1    
# 同理,找出数据集X2中第二列(即'Spending Score (1-100)')的最小值和最大值,并分别减1和加1来扩展网格的边界。  
y_min, y_max = X2[:, 1].min() - 1, X2[:, 1].max() + 1    
# 使用numpy的meshgrid函数生成一个二维网格。其中,np.arange(x_min, x_max, h)和np.arange(y_min, y_max, h)分别生成x和y轴上的点。  
# meshgrid函数将这两个一维数组转换为两个二维数组xx和yy,它们表示网格上的所有点的坐标。  
xx, yy = np.meshgrid(np.arange(x_min, x_max, h), np.arange(y_min, y_max, h))    
# 将xx和yy的二维数组展平(ravel)成一维数组,并使用np.c_将它们组合成一个二维数组,其中每一行代表网格上的一个点('Annual Income (k$)'和'Spending Score (1-100)'的坐标)。  
# 然后,使用KMeans对象(algorithm)的predict方法预测这些点所属的簇。  
# Z2将是一个一维数组,其中包含了网格上每个点所属的簇的标签。  
Z2 = algorithm.predict(np.c_[xx.ravel(), yy.ravel()])
# 创建一个新的图形窗口,编号为1,并设置其大小为15x7英寸  
plt.figure(1 , figsize = (15 , 7) )
# 清除当前图形窗口的内容(如果有的话),确保在干净的画布上绘图  
plt.clf()
# 由于Z2原本是一维数组,它包含了每个网格点所属的簇的标签。我们需要将其重新整形为与xx相同的形状,  
# 以便可以使用imshow来绘制聚类结果。  
Z2 = Z2.reshape(xx.shape)
# 使用imshow函数绘制Z2数组,即聚类结果。interpolation='nearest'表示使用最近邻插值法。  
# extent参数定义了图像在x和y轴上的扩展范围,与xx和yy的边界对应。  
# cmap=plt.cm.Pastel2设置了颜色映射,这里使用了Pastel2配色方案。  
# aspect='auto'允许自动调整x和y轴的比例,以保持数据的真实比例。  
# origin='lower'表示图像的原点在左下角。  
plt.imshow(Z2 , interpolation='nearest', extent=(xx.min(), xx.max(), yy.min(), yy.max()),cmap = plt.cm.Pastel2, aspect = 'auto', origin='lower')
# 绘制原始数据点。这里不能直接使用字符串'Annual Income (k$)'和'Spending Score (1-100)'作为x和y的值,  
# 而是应该从数据框df中提取这些列的实际值。c=labels2表示使用labels2数组中的值来为数据点着色,  
# s=200设置了数据点的大小。 
plt.scatter( x = 'Annual Income (k$)' ,y = 'Spending Score (1-100)' , data = df , c = labels2 , s = 200 )
# 绘制聚类中心(质心)。x和y分别使用centroids2数组的第一列和第二列的值。  
# s=300设置了质心点的大小,c='red'表示质心点用红色绘制,alpha=0.5设置了点的透明度。 
plt.scatter(x = centroids2[: , 0] , y =  centroids2[: , 1] , s = 300 , c = 'red' , alpha = 0.5)
# 设置y轴的标签为'Spending Score (1-100)',设置x轴的标签为'Annual Income (k$)'。  
# 注意这里使用逗号分隔了两个函数调用,是Python中一种常见的在一行内写多个语句的方式。 
plt.ylabel('Spending Score (1-100)') , plt.xlabel('Annual Income (k$)')
plt.show()

年收入和支出分数的可视化KMeans聚类的结果:
在这里插入图片描述

# 注意:iloc[: , :]在这里是多余的,因为它实际上选择了所有的行和列,但在这里它不会造成错误。  
X3 = df[['Age', 'Annual Income (k$)', 'Spending Score (1-100)']].iloc[:, :].values    
# 创建一个空列表,用于存储每次KMeans聚类后的惯性(即簇内距离的平方和)  
inertia = []    
# 循环遍历从1到10的整数(包括1但不包括11),代表要形成的簇的数量  
for n in range(1, 11):  # 创建一个KMeans对象,设置n_clusters为当前的n值(即簇的数量),以及其他一些参数  algorithm = (KMeans(n_clusters=n, init='k-means++', n_init=10, max_iter=300,   tol=0.0001, random_state=111, algorithm='elkan'))       # 使用KMeans对象对X3数据进行拟合(即进行聚类)  algorithm.fit(X3)        # 获取这次聚类的惯性,并将其添加到inertia列表中  inertia.append(algorithm.inertia_)    
# 现在inertia列表包含了从1到10个簇时的惯性值,可以用于后续的图形化显示或选择最佳簇数(即“肘部方法”)
# 创建一个新的图形窗口,编号为1,并设置其大小为15x6英寸  
plt.figure(1, figsize=(15, 6))    
# 使用plot函数绘制散点图,表示簇的数量与对应的惯性值。  
# np.arange(1, 11)生成了一个从1到10的整数数组,代表簇的数量。  
# inertia是之前计算得到的惯性值列表。  
# 'o'指定了点的样式为圆圈。  
plt.plot(np.arange(1, 11), inertia, 'o')    
# 再次使用plot函数绘制线图,以连接上述散点,这样可以更清晰地看到惯性值随簇数量增加的变化趋势。  
# '-'指定了线的样式为实线。  
# alpha=0.5设置了线的透明度,使散点图更加突出。  
plt.plot(np.arange(1, 11), inertia, '-', alpha=0.5)    
# 设置x轴的标签为'Number of Clusters',表示簇的数量。  
# 注意这里使用了逗号分隔了两个函数调用,是Python中一种常见的在一行内写多个语句的方式。  
plt.xlabel('Number of Clusters'), plt.ylabel('Inertia')    
# 显示图形窗口  
plt.show()

KMeans聚类中聚类数量与惯性值之间的关系图:
在这里插入图片描述

# 创建一个KMeans对象,设置其参数来初始化KMeans聚类算法。  
# 参数说明如下:  
# n_clusters: 指定的簇数量,这里设置为6,表示数据将被分为6个簇。  
# init: 初始化方法,'k-means++' 是一种用于选择初始质心集合的算法,它通常可以产生比随机初始化更好的结果。  
# n_init: 使用不同的初始质心设置运行KMeans算法的次数,选择最佳结果(即最小惯性)的那次。这里设置为10,意味着算法会运行10次,每次使用不同的初始质心。  
# max_iter: 单次运行的迭代最大次数,即算法会尝试优化簇质心的最大次数。这里设置为300。  
# tol: 容忍的最小优化值,当(惯性)改进低于这个值时,算法将停止迭代。这里设置为0.0001。  
# random_state: 随机数生成器的种子,用于初始化质心。设置随机种子可以确保结果的可重复性。这里设置为111。  
# algorithm: KMeans算法的实现,'elkan'是一种更高效的实现,尤其是在大数据集上。  
algorithm = (KMeans(n_clusters=6, init='k-means++', n_init=10, max_iter=300,   tol=0.0001, random_state=111, algorithm='elkan'))    
# 使用KMeans对象对X3数据进行拟合(即进行聚类),将数据划分为n_clusters指定的簇数(在这里是6个簇)。  
# 这个过程会找到每个簇的质心,并将每个数据点分配给最近的质心所代表的簇。  
algorithm.fit(X3)    
# labels3变量用于存储每个数据点所属的簇的标签。  
# 这些标签是整数,范围从0到n_clusters-1(在这里是从0到5)。  
labels3 = algorithm.labels_    
# centroids3变量用于存储每个簇的质心(即簇的中心点)。  
# 它是一个二维数组,其中每一行代表一个簇的质心,每一列代表一个特征(在这里是Age、Annual Income (k$)和Spending Score (1-100))。  
centroids3 = algorithm.cluster_centers_
# 将KMeans算法生成的簇标签(labels3)添加到原始数据框df中作为一个新的列'label3'。  
df['label3'] = labels3    
# 创建一个3D散点图的trace对象,使用plotly的go.Scatter3d类。  
trace1 = go.Scatter3d(  # 设置x轴的数据为df中的'Age'列。  x= df['Age'],  # 设置y轴的数据为df中的'Spending Score (1-100)'列。  y= df['Spending Score (1-100)'],  # 设置z轴的数据为df中的'Annual Income (k$)'列。  z= df['Annual Income (k$)'],  # 设置模式为'markers',即只显示标记点,不显示连线。  mode='markers',  # 设置标记点的属性。  marker=dict(  # 设置标记点的颜色为df中的'label3'列,即每个簇使用不同的颜色。  color = df['label3'],   # 设置标记点的大小为20。  size= 20,  # 设置标记点的边界线属性。  line=dict(  # 设置边界线的颜色与标记点颜色相同(即与簇标签对应)。  color= df['label3'],  # 设置边界线的宽度为12。  width= 12  ),  # 设置标记点的透明度为0.8。  opacity=0.8  )  
)    
# 创建一个包含trace1的数据列表。  
data = [trace1]    
# 创建一个布局对象,用于设置图表的整体布局。  
layout = go.Layout(  # (注释掉的代码)设置图表的边距,但这里被注释掉了,所以图表将使用默认边距。  # title= 'Clusters', # 标题设置为'Clusters'  # 设置图表的标题为'Clusters'。  title= 'Clusters',  # 设置3D场景的属性。  scene = dict(  # 设置x轴的属性,标题为'Age'。  xaxis = dict(title  = 'Age'),  # 设置y轴的属性,标题为'Spending Score'。  yaxis = dict(title  = 'Spending Score'),  # 设置z轴的属性,标题为'Annual Income'。  zaxis = dict(title  = 'Annual Income')  )  
)    
# 创建一个Figure对象,并传入数据和布局。  
fig = go.Figure(data=data, layout=layout)    
# 使用plotly的offline模块来在本地显示图表。注意:py.offline.iplot是一个较旧的plotly API调用,  
# 在新版本的plotly中,可能需要使用其他方法(如plotly.offline.plot或直接在Jupyter Notebook中使用fig.show())。  
# 这里假设您已经正确地设置了plotly的offline模式或者在一个支持的环境下运行此代码。  
py.offline.iplot(fig)

年龄、年收入和支出分数的可视化KMeans聚类的结果:
在这里插入图片描述
最后我以最后一次可视化的结果为例,将分类数值映射到原数据,直观的反映聚类结果。

在这里插入图片描述
最后一列就是我们的聚类结果了,从最后一次聚类可视化结果可以看出,有5个红色的质心,共有5个簇(也就是按超市会员的不同特征分成5类人)。K-Means算法在商场顾客细分领域的应用是一个典型的案例,它可以帮助商场管理者更好地理解顾客群体的特征和偏好,从而制定更加精准的营销策略。

创作不易,点赞、收藏、评论、转发,评论区留邮箱免费拿数据集。

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/diannao/31199.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

Spring Boot集成tablesaw插件快速入门

1 什么是tablesaw? Tablesaw是一款Java的数据可视化库,主要包括两部分: 数据解析库,主要用于加载数据,对数据进行操作(转化,过滤,汇总等),类比Python中的Pandas库; 数据…

苹果cms10影视网整站源码下载/苹果cms模板MXone Pro自适应影视电影网站模板

下载地址:苹果cms10影视网整站源码下载/苹果cms模板MXone Pro自适应影视电影网站模板 模板带有夜间模式、白天晚上自动切换,有观影记录、后台设置页。全新UI全新框架,加载响应速度更快,seo更好,去除多余页面优化代码。…

从零开始搭建创业公司全新技术栈解决方案

从零开始搭建创业公司全新技术栈解决方案 关于猫头虎 大家好,我是猫头虎,别名猫头虎博主,擅长的技术领域包括云原生、前端、后端、运维和AI。我的博客主要分享技术教程、bug解决思路、开发工具教程、前沿科技资讯、产品评测图文、产品使用体…

Ollma本地大模型沉浸式翻译【403报错解决】

最终效果 通过Chrome的 沉浸式翻译 插件,用OpenAI通用接口调用本地的Ollma上的模型,实现本地的大模型翻译文献。 官方文档指导的Ollama的配置:一定要配置环境变量,否则会出现【403报错】

GoogLeNet(InceptionV3)模型算法

GoogLeNet 团队在给出了一些通用的网络设计准则,以期望在不提高网络参数 量的前提下提升网络的表达能力: 避免特征图 (feature map) 表达瓶颈:从理论上讲,尺寸 (seize) 才包含了相关结构等重要因素,维度(channel) 仅仅…

torch.optim 之 Algorithms (Implementation: for-loop, foreach, fused)

torch.optim的官方文档 官方文档中文版 一、Implementation torch.optim的官方文档在介绍一些optimizer Algorithms时提及它们的implementation共有如下三个类别:for-loop, foreach (multi-tensor), and fused。 Chat-GPT对这三个implementation的解释是&#xf…

账号和权限的管理

文章目录 管理用户账号和组账号用户账号的分类超级用户普通用户程序用户 UID(用户id)和(组账号)GIDUID用户识别号GID组标识号 用户账号文件添加用户账号设置/更改用户口令 管理用户账号和组账号 用户账号的分类 超级用户 root 用户是 Linux 操作系统中默认的超级…

第N5周:调用Gensim库训练Word2Vec模型

🍨 本文为🔗365天深度学习训练营 中的学习记录博客🍖 原作者:K同学啊 | 接辅导、项目定制🚀 文章来源:K同学的学习圈子 目录 本周任务: 1.安装Gensim库 2.对原始语料分词 3.停用词 4.训练Woed2Vec模型 …

办展览如何盈利?论办展的商业模式

想要弄清楚办展览怎么赚钱这个问题,我可以来说说。 首先来说说展览收益的大头:门票收入。 这个其实是可以大致预测的。简单来说,就是用流量乘以到店率。 但别忘了,这背后得有合适的展览定位、方便的展览场地和合理的票价。 说…

小林图解系统-三、操作系统结构

Linux 内核 vs Windows 内核 内核 作为应用连接硬件设备的桥梁,保证应用程序只需要关心与内核交互,不需要关心硬件的细节 内核具备四个基本能力: 管理进程、线程,决定哪个进程、线程使用CPU,也就是进程调度的能力&a…

Linux——ansible关于“文件操作”的模块

修改文件并将其复制到主机 一、确保受管主机上存在文件 使用 file 模块处理受管主机上的文件。其工作方式与 touch 命令类似,如果不存在则创建一个空文件,如果存在,则更新其修改时间。在本例中,除了处理文件之外,Ansi…

华为设备SSH远程访问配置实验简述

一、实验需求: 1、AR1模拟电脑SSH 访问AR2路由器。 二、实验步骤: 1、AR1和AR2接口配置IP,实现链路通信。 2、AR2配置AAA模式 配置用户及密码 配置用户访问级别 配置用户SSH 访问服务 AR2配置远程服务数量 配置用户远程访问模式为AAA 配置允许登录接入用…

【问题记录】Ubuntu提示: “E: 软件包 gcc 没有可安装候选“

Ubuntu提示: "E: 软件包 gcc 没有可安装候选" 一,问题现象二,问题原因&解决方法 一,问题现象 在虚拟机Ubuntu中进行安装gcc命令时报错:“E: 软件包 gcc 没有可安装候选”: 二,问题原因&解决方法 …

性能测试-性能监控分析与调优(三)《实战》

性能监控 使用命令监控 cpu瓶颈分析 top命令 在进行性能测试时使用top命令,界面如下 上图可以看出 CPU 概况区: %Cpu(s): us(用户进程占用CPU的百分比), 和 sy(系统进程占用CPU的百分比) 的数值很高…

代码随想录-Day36

452. 用最少数量的箭引爆气球 有一些球形气球贴在一堵用 XY 平面表示的墙面上。墙面上的气球记录在整数数组 points ,其中points[i] [xstart, xend] 表示水平直径在 xstart 和 xend之间的气球。你不知道气球的确切 y 坐标。 一支弓箭可以沿着 x 轴从不同点 完全垂…

Linux--视频推流及问题

方案一: mjpg-streamer,它运行在ARM板上 在手机上使用浏览器直接观看视频 方案二: 推流端(Fmpeg)--rtmp-->Nginx(流媒体服务器)--rtmp/httpflv/hls-->浏览器、播放器 此篇文章记录方案二的具体细…

Meta悄咪咪的发布多款AI新模型

大模型技术论文不断,每个月总会新增上千篇。本专栏精选论文重点解读,主题还是围绕着行业实践和工程量产。若在某个环节出现卡点,可以回到大模型必备腔调或者LLM背后的基础模型重新阅读。而最新科技(Mamba,xLSTM,KAN)则…

自定义线程池

自定义线程池需要什么 需要哪些类 MyTask /*** 自定义线程池任务* 要求每个线程有自己的编号* 线程的执行时间为0.2s*/ Data public class MyTask implements Runnable{private int id;MyTask(int id) {this.id id;}Overridepublic void run() {String name Thread.curren…

基于Java的家政服务管理平台

开头语:你好呀,我是计算机学姐码农小野!如果有相关需求,可以私信联系我。 开发语言:Java 数据库:MySQL 技术:B/S结构,SpringBoot框架 工具:MyEclipse,Nav…

网络编程5----初识http

1.1 请求和响应的格式 http协议和前边学过的传输层、网络层协议不同,它是“一问一答”形式的,所以要分为请求和响应两部分看待,同时,请求和响应的格式是不同的,我们来具体介绍一下。 1.1.1 请求 在介绍请求之前&…