python - 在 Python 中对 CSV 进行排序

Question

我假设使用 Python 在多个文本/数字字段上对 CSV 文件进行排序是一个已经解决的问题。但是我在任何地方都找不到任何示例代码，除了专注于对日期字段进行排序的特定代码。

如何在多个字段上按顺序对相对较大的 CSV 文件(数万行)进行排序？

Python 代码示例将不胜感激。

Answer 1

Python 的排序只在内存中工作；然而，在现代机器上，数以万计的代码行应该可以轻松地放入内存中。所以:

import csv

def sortcsvbymanyfields(csvfilename, themanyfieldscolumnnumbers):
  with open(csvfilename, 'rb') as f:
    readit = csv.reader(f)
    thedata = list(readit)
  thedata.sort(key=operator.itemgetter(*themanyfieldscolumnnumbers))
  with open(csvfilename, 'wb') as f:
    writeit = csv.writer(f)
    writeit.writerows(thedata)