处理csv文件时,经常用到多级或者多属性排序,我们可以使用如下方式轻松搞定多级排序。
Python 2.x使用如下代码:先按照第一列即x[0]升序排序,再按照第二列即x[0]升序排序。如需降序排列,需在sorted函数体内加上reverse = True。
import sys, csv , operator
data = csv.reader(open
('C:\test.csv'),delimiter=',')
sortedlist = sorted(data, key = lambda x: (x[0], int(x[1])))
with open("C:\result_test.csv", "wb") as f:
fileWriter = csv.writer(f, delimiter=',')
for row in sortedlist:
fileWriter.writerow(row)
f.close()
Python 3.x使用如下代码:先按照第一列即x[0]升序排序,再按照第二列即x[0]升序排序。如需降序排列,需在sorted函数体内加上reverse = True。
import sys, csv , operator
data = csv.reader(open('C:\\test.csv'),delimiter=',')
sortedlist = sorted(data, key = lambda x: (x[0], int(x[1])))
with open("C:\\result_test.csv", "w", newline = '') as f:
fileWriter = csv.writer(f, delimiter=',')
for row in sortedlist:
fileWriter.writerow(row)
f.close()