Python,转置列表并写入CSV文件

时间:2012-05-13 17:52:30

标签: python list csv

我需要使用python写入csv文件,每个迭代器项应该以新行开头。 所以我使用的分隔符是“\ n”。 写完每个列表后,下一个列表应该从下一个单元格写入。 如下所示:

 lol = [[1,2,3],[4,5,6]]

csv就像:

1 4
2 5
3 6

我尝试过:

file = open("test.csv", "wb")
fileWriter = csv.writer(file , delimiter='\n',quotechar='|', quoting=csv.QUOTE_MINIMAL)
spamWriter.writerow([1,2,3])
spamWriter = csv.writer(file , delimiter=',',quotechar='|', quoting=csv.QUOTE_MINIMAL)
spamWriter.writerow([4,5,6])
file.close()

结果如下:

 1
 2
 3
 4 5 6

使用csv模块如何获得如下输出:

 1 4 
 2 5
 3 6

此处空格表示csv文件中的逗号。

感谢。

3 个答案:

答案 0 :(得分:13)

不使用zip,你可以这样做:

import csv

lol = [[1,2,3],[4,5,6],[7,8,9]]
item_length = len(lol[0])

with open('test.csv', 'wb') as test_file:
  file_writer = csv.writer(test_file)
  for i in range(item_length):
    file_writer.writerow([x[i] for x in lol])

这将输出到test.csv:

1,4,7
2,5,8
3,6,9

答案 1 :(得分:9)

首先使用zip()

转置您的输入
>>> zip(*lol)
[(1, 4), (2, 5), (3, 6)]

然后将其传递给csw.writer例如

with open("test.csv", "wb") as f:
    fileWriter = csv.writer(f, delimiter=',',quotechar='|', quoting=csv.QUOTE_MINIMAL)
    for row in zip(*lol):
        fileWriter.writerow(row)

...结果是:

$ cat test.csv 
1,4
2,5
3,6

答案 2 :(得分:3)

如果您使用的是Python3,则需要以“wt”文本格式打开文件,而csv以上的文件可以writerows,可以用来一次写入所有内容。这是一个例子:

data=[("test", "value1", "value2"), ("test2", "value3", "value4")]
with open('my.csv','wt') as out:
   csv_out=csv.writer(out)
   csv_out.writerows(data)

我刚刚注意到问题是如何转换列表,这是一个单独的步骤,我就是这样做的:

lol = [[1,2,3],[4,5,6]]
data = zip(lol[0],lol[1])