Question

我有一个csv文件，如下所示

date                       mse                                                  
2018-02-11                 14.34
2018-02-12                 7.24
2018-02-13                 4.5
2018-02-14                 3.5
2018-02-16                 12.67
2018-02-21                 45.66
2018-02-22                 15.33
2018-02-24                 98.44
2018-02-26                 23.55
2018-02-27                 45.12
2018-02-28                 78.44
2018-03-01                 34.11
2018-03-05                 23.33
2018-03-06                 7.45
...                        ...

现在我想为mse值获取两个聚类，以便我知道哪个聚类及其平均值是什么值。

既然我除了mse之外没有任何其他值集（我必须提供X和Y），我想只使用mse值来获得ak意味着集群。现在对于另一组值，我将其作为范围传递，其大小与mse值的大小相同。这就是我所做的

from sklearn.cluster import KMeans
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

df = pd.read_csv("generate_csv/all_data_device.csv", parse_dates=["date"])
f1 = df['mse'].values
# generate another list
f2 = list(range(0, len(f1)))
X = np.array(list(zip(f1, f2)))
kmeans = KMeans(n_clusters=2).fit(X)
labels = kmeans.predict(X)
# Centroid values
centroids = kmeans.cluster_centers_
#print(centroids)

fig = plt.figure()
ax = Axes3D(fig)
ax.scatter(X[:, 0], X[:, 1], c=labels)
ax.scatter(centroids[:, 0], centroids[:, 1], marker='*', c='#050505', s=1000)
plt.title('K Mean Classification')
plt.show()

如何使用mse值来获取k表示集群？我知道函数'reshape（）'但不太确定如何使用它？

Answer 1

演示：

In [29]: kmeans = KMeans(n_clusters=2)

In [30]: df['label'] = kmeans.fit_predict(df[['mse']])
# NOTE:                     ---->            ^     ^

In [31]: df
Out[31]:
          date    mse  label
0   2018-02-11  14.34      0
1   2018-02-12   7.24      0
2   2018-02-13   4.50      0
3   2018-02-14   3.50      0
4   2018-02-16  12.67      0
5   2018-02-21  45.66      0
6   2018-02-22  15.33      0
7   2018-02-24  98.44      1
8   2018-02-26  23.55      0
9   2018-02-27  45.12      0
10  2018-02-28  78.44      1
11  2018-03-01  34.11      0
12  2018-03-05  23.33      0
13  2018-03-06   7.45      0

绘制：

In [64]: ax = df[df['label']==0].plot.scatter(x='mse', y='label', s=50, color='white', edgecolor='black')

In [65]: df[df['label']==1].plot.scatter(x='mse', y='label', s=50, color='white', ax=ax, edgecolor='red')
Out[65]: <matplotlib.axes._subplots.AxesSubplot at 0xfa42be0>

In [66]: plt.scatter(kmeans.cluster_centers_.ravel(), [0.5]*len(kmeans.cluster_centers_), s=100, color='green', marker='*')
Out[66]: <matplotlib.collections.PathCollection at 0xfabf208>

如何获得k意味着1D数据的集群？

1 个答案: