温馨提示×

Pandas中怎么实现数据的聚类分析

小亿
89
2024-05-13 11:15:52
栏目: 编程语言

在Pandas中实现数据的聚类分析,一般需要借助于scikit-learn库中的聚类算法,例如KMeans。以下是一个简单的示例代码:

import pandas as pd
from sklearn.cluster import KMeans

# 创建一个包含数据的DataFrame
data = {'Feature1': [1, 2, 3, 4, 5],
        'Feature2': [5, 4, 3, 2, 1]}
df = pd.DataFrame(data)

# 初始化KMeans模型,指定聚类数量为2
kmeans = KMeans(n_clusters=2)

# 对数据进行聚类分析
df['Cluster'] = kmeans.fit_predict(df)

# 打印聚类结果
print(df)

在上面的示例中,首先创建了一个包含两个特征的DataFrame,然后使用KMeans算法对数据进行聚类分析,最终将聚类结果保存在DataFrame中并打印出来。通过这种方式,可以在Pandas中实现数据的聚类分析。

0