如何在python中使数据类型为整数

Question

我想使用分层聚类对数据进行聚类，我希望结果是数据类型为整数而不是浮点

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

Answer 1

使用列表理解：

int_labels = [int(label) for label in cluster.labels_]