排序时数据文件头消失

Question

我有一个包含多行数据的 csv 文件。第一行是列的标题。 我想按一些参数（特别是第一列）对数据进行排序，但当然要保留标题。 当我执行以下操作时，标题完全消失并且不包含在输出文件中。 谁能建议如何保留标题但跳过它并对其余行进行排序？ （对于 fwiw，第一列是数字和字母的混合）。

谢谢！

这是我的代码：

import csv
import operator

sankey = open('rawforsankey.csv', "rb")
raw_reader = csv.reader(sankey)


raw_data = []
for row in raw_reader:
    raw_data.append(row)


raw_data_sorted = sorted(raw_data, key=operator.itemgetter(0))

myfiletest = open('newfiletest.csv', 'wb')
wr = csv.writer(myfiletest,quoting = csv.QUOTE_ALL)

wr.writerows(raw_data_sorted)


sankey.close()
myfiletest.close()

编辑：应该提到我在代码中尝试了这种变化：

raw_data_sorted = sorted(raw_data[1:], key=operator.itemgetter(0))

但得到了相同的结果

Answer 1

您对所有数据（包括标题）进行了排序，这意味着它仍然存在，但可能在结果输出的中间某处。

这就是您如何在一列上对 CSV 进行排序，同时保留标题：

import csv
import operator

with open('rawforsankey.csv', "rb") as sankey:
    raw_reader = csv.reader(sankey)
    header = next(raw_reader, None)
    sorted_data = sorted(raw_reader, operator.itemgetter(0))

with open('newfiletest.csv', 'wb') as myfiletest:
    wr = csv.writer(myfiletest, quoting=csv.QUOTE_ALL)
    if header:
        wr.writerow(header)
    wr.writerows(sorted_data)

请记住，排序是按字典顺序完成的，因为所有列都是字符串。例如，10 排在9 之前。例如，如果您的数据是数字，请使用更具体的排序键。