Ciao a tutti, sono nuova e sto seguendo un corso di python all'università, devo calcolare media, mediana e quantili di un cluster rispetto ad una variabil, ho costruito il codice così...
ma non riesco a stamparlo (la parte in grassetto mi da errore), qualcuno sa dirmi perché?
clusterer = KMeans(n_clusters=n_clusters, random_state=10)
cluster_labels = clusterer.fit_predict(df)
def create_cluster_to_txt(variable1=1, variable2=4):
clusterer = KMeans(n_clusters=n_clusters, random_state=10)
cluster_labels = clusterer.fit_predict(df)
m,s,q1,q2,q3 = [],[],[],[],[]
momenti = []
for i in set(cluster_labels):
index = cluster_labels == i
media = np.mean(df[index,variable1])
varianza = np.std(df[index,variable1])
primo_quartile = np.quantile(df[index,variable1],0.25)
secondo_quartile = np.quantile(df[index,variable1],0.5)
terzo_quartile = np.quantile(df[index,variable1],0.75)
m.append(media)
s.append(varianza)
q1.append(primo_quartile)
q2.append(secondo_quartile)
q3.append(terzo_quartile)
print "Cluster {i}\n Media:{media}, Varianza: {varianza}, Quartili: {uno},{due},{tre}".format(i=i, media=round(media,3),varianza=round(varianza,3),uno=round(primo_quartile,3), due=round(secondo_quartile,3), tre=round(terzo_quartile,3) )
momenti.append(m)
momenti.append(s)
momenti.append(q1)
momenti.append(q2)
momenti.append(q3)
np.savetxt('test.txt', momenti)