python - 在 Python 中以不同的顺序写出 CSV 列

Question

我意识到这与 question 非常相似.但是，我有一个 CSV 文件，它总是采用相同的格式，我需要以不同的顺序用列写出，以便将其向下移动到数据处理管道。如果我的 csv 文件包含这样的标题和数据:

Date,Individual,Plate,Sample,test,QC
03312011,Indiv098,P342,A1,deep,passed
03312011,Indiv113,P352,C3,deep,passed

我如何写出与原始输入 csv 具有相同列但按以下顺序排列的 csv 文件:

test,QC,Plate,Sample
deep,passed,P342,A1
deep,passed,P352,C3

我最初的想法是做这样的事情:

f = open('test.csv')
lines = f.readlines()
for l in lines:
    h = l.split(",")
    a, b, c, d, e, f  = h
    for line in h:
        print e, f, c, d, 

Answer 1

如果输入文件或输出文件每次都不具有相同布局的可能性很小，那么这里有一个更通用的获取“reorderfunc”的方法:

writenames = "test,QC,Plate,Sample".split(",") # example
reader = csv.reader(input_file_handle)
writer = csv.writer(output_file_handle)
# don't forget to open both files in binary mode (2.x)
# or with `newline=''` (3.x)
readnames = reader.next()
name2index = dict((name, index) for index, name in enumerate(readnames))
writeindices = [name2index[name] for name in writenames]
reorderfunc = operator.itemgetter(*writeindices)
writer.writerow(writenames)
for row in reader:
    writer.writerow(reorderfunc(row))