Python K-均值聚类实验

原创

于 2024-08-01 20:01:28 发布 · 242 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#均值算法 #聚类 #算法

利用某地2021年、2022年的3月、6月、9月、12月份的逐日平均气温T、相对湿度RH、位势高度Z850的再分析资料构建训练样本X=[T,RH,Z850]T。利用K-Means算法实现对基于这三个要素构成的特征空间的聚类分析。

1、利用Scikit-Learn实现K-Means聚类

取K=2，3，4，可视化聚类效果

代码如下：

import numpy as np
from sklearn.cluster import KMeans
from sklearn.datasets import make_blobs
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

data = np.load('data_experiment3.npy')
n_clusters = [4]

def fit_plot_kmean_model(n_clusters,X):
    kmean = KMeans(n_clusters=n_cluster