python - 如何在 csv 文件中转置数据集？

Question

例如，我想改造:

Name,Time,Score
Dan,68,20
Suse,42,40
Tracy,50,38

进入:

Name,Dan,Suse,Tracy
Time,68,42,50
Score,20,40,38

编辑:原始问题错误地使用了术语“转置”。

Answer 1

如果整个文件内容适合内存，您可以使用

import csv
from itertools import izip
a = izip(*csv.reader(open("input.csv", "rb")))
csv.writer(open("output.csv", "wb")).writerows(a)

您基本上可以将 zip() 和 izip() 视为转置操作:

a = [(1, 2, 3),
     (4, 5, 6),
     (7, 8, 9)]
zip(*a)
# [(1, 4, 7),
#  (2, 5, 8),
#  (3, 6, 9)]

izip() 避免立即复制数据，但基本上会这样做。