我想在CSV文件中编写聚类结果。以下代码是k-means聚类的一小部分,我用它来编写聚类结果在CSV文件中。此示例使用excel。我首先感兴趣的是将数据输入我的程序并在CSV文件中获得答案:
with open('some3.csv', 'w') as test_file:
csv_writer = csv.writer(test_file,dialect='excel',delimiter=',')
for y,c in enumerate(clusters):
csv_writer.writerows("cluster %d:" % y + "\t%s" % cat[i] for i in c)
我希望它是5个集群,但不是得到:
cluster 0,value1
cluster 0,value2
cluster 0,value3
cluster 1 value1
cluster 1 value2
.
.
.
我的CSV文件中有:
c l u s t e r 0 v a l u e 1
c l u s t e r 0 v a l u e 2
.
.
.
。 如何加入角色?有什么建议吗?
答案 0 :(得分:0)
csv_writer.writerows
需要list
个列表,你给它一个字符串,这是一个字符列表。
试试这个:
csv_writer.writerow(["cluster %d:" % y] + cat[i] for i in c)