как сделать тип данных целочисленным в 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_]