我想生成一个.csv的二维平面分布的人工数据集文件,想把代码中生成的值输出到.csv文件中,每一行用逗号隔开,有两列,形如:
1,2
第一列代表x的值,第二列代表y的值。
代码如下:
from sklearn.datasets import make_blobs
from matplotlib import pyplot
data,target=make_blobs(n_samples=100,n_features=2,centers=3,cluster_std=[1.0,3.0,2.0])
pyplot.scatter(data[:,0],data[:,1],c=target);
pyplot.show()
上面这段代码只能显示出所生成的图片,我想把数据也输出来但是不对,我改的代码如下:
from sklearn.datasets import make_blobs
from matplotlib import pyplot
import sys
data,target=make_blobs(n_samples=100,n_features=2,centers=3,cluster_std=[1.0,3.0,2.0])
pyplot.scatter(data[:,0],data[:,1],c=target);
output=sys.stdout
outputfile=open(r'E:Datasyntheticdataset.csv','w')
sys.stdout=outputfile
pyplot.show()
outputfile.close()
sys.stdout=output
经过修改能生成syntheticdataset.csv这个文件但是里面并没有我想要的数据,请问下大神们该如何修改?
相关分类