按列排序CSV

Question

我想按照日期对一个CSV表格进行排序。一开始这看起来是个简单的任务：

import sys
import csv

reader = csv.reader(open("files.csv"), delimiter=";")

for id, path, title, date, author, platform, type, port in reader:
    print date

我使用了Python的CSV模块来读取这个结构的文件：

id;file;description;date;author;platform;type;port

• 日期采用ISO-8601格式，所以我可以很容易地进行排序，不需要额外解析，比如：2003-04-22。
• 我想按日期排序，最新的记录排在前面。
• 我该如何将这个读取器转换成一个可以排序的数据结构呢？我想如果努力的话，可以创建一个日期列表：datelist += date，然后分割和排序。不过我还得重新识别CSV表格中的完整条目。这不仅仅是对一堆东西进行排序。
• csv模块似乎没有内置的排序功能。

最理想的解决方案是有一个CSV客户端，像数据库一样处理这个文件。但我没有找到这样的东西。

希望有人能分享一些不错的排序技巧；)

Answer 1

这个读取器就像一个生成器。在一个包含一些假数据的文件上：

>>> import sys, csv
>>> data = csv.reader(open('data.csv'),delimiter=';')
>>> data
<_csv.reader object at 0x1004a11a0>
>>> data.next()
['a', ' b', ' c']
>>> data.next()
['x', ' y', ' z']
>>> data.next()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration

使用 operator.itemgetter，就像 Ignacio 建议的那样：

>>> data = csv.reader(open('data.csv'),delimiter=';')
>>> import operator
>>> sortedlist = sorted(data, key=operator.itemgetter(2), reverse=True)
>>> sortedlist
[['x', ' y', ' z'], ['a', ' b', ' c']]

Answer 2

要按多个列进行排序（先按column_1排序，然后再按column_2排序）

with open('unsorted.csv',newline='') as csvfile:
    spamreader = csv.DictReader(csvfile, delimiter=";")
    sortedlist = sorted(spamreader, key=lambda row:(row['column_1'],row['column_2']), reverse=False)


with open('sorted.csv', 'w') as f:
    fieldnames = ['column_1', 'column_2', column_3]
    writer = csv.DictWriter(f, fieldnames=fieldnames)
    writer.writeheader()
    for row in sortedlist:
        writer.writerow(row)

Answer 3

因为在这一列中，'date'的索引是3，

import operator
sortedlist = sorted(reader, key=operator.itemgetter(3), reverse=True)

或者可以使用lambda表达式

sortedlist = sorted(reader, key=lambda row: row[3], reverse=True)