【发布时间】:2019-09-25 12:10:47
【问题描述】:
我想使用层次聚类对数据进行聚类,我希望结果是数据类型是整数而不是浮点数
import pandas as pd
import csv
#import csv
#from nltk.cluster import KMeansClusterer, euclidean_distance
dataset =pd.read_csv('vector.csv')
X = dataset
from sklearn.cluster import AgglomerativeClustering
cluster = AgglomerativeClustering(n_clusters=5, affinity='euclidean', linkage='ward')
cluster.fit_predict(X)
#print(cluster.labels_)
file=open('data_hirarki/hirarki.csv','a')
tulis=csv.writer(file,delimiter='\n',lineterminator='\n')
tulis.writerows([cluster.labels_])
file.close()
我该怎么办?
结果像
1.0, 2.0, etc
我想要 1、2,3
【问题讨论】:
标签: python csv scikit-learn floating-point integer