如何通过拆分列将列表保存为csv

时间:2017-03-21 19:34:44

标签: python csv machine-learning dataset pycharm

我有脑电图数据集列表。它们看起来像这样:

['-0.9148', '-1.7609', '0.8441', '-3.0872', '-3.3303', '-2.5054', '1.5679', '-4.6378', '-3.5720', '-3.3940']

我想将它们保存为CSV格式

['-0.9148', '-1.7609', '0.8441', '-3.0872']
['-3.3303', '-2.5054', '1.5679', '-4.6378']
['-0.9148', '-2.5054', '-3.5720','-3.3940']

但是,我将它们保存为列表中的样子。

1 个答案:

答案 0 :(得分:1)

你可以这样做:

>>> x = ['-0.9148', '-1.7609', '0.8441', '-3.0872', '-3.3303', '-2.5054', '1.5679', '-4.6378', '-3.5720', '-3.3940']
>>> zip(*[x[i::3] for i in range(4)])
[('-0.9148', '-1.7609', '0.8441', '-3.0872'), ('-3.0872', '-3.3303', '-2.5054', '1.5679'), ('1.5679', '-4.6378', '-3.5720', '-3.3940')]

它基本上将第i个元素元素作为你需要的元素,给你n个第一,第二,第n个元素的向量,然后在结果上调用zip。

然后写入csv:

with open('filename.csv', 'wb') as csvfile:
    w = csv.writer(csvfile)
    w.writerows(zip(*[x[i::3] for i in range(4)]))