python如何给csv里的数据排序 python给csv里的数据排序代码示例

python如何给csv里的数据排序 python给csv里的数据排序代码示例

本篇文章小编给大家分享一下python给csv里的数据排序代码示例，文章介绍的很详细，小编觉得挺不错的，现在分享给大家供大家参考，有需要的小伙伴们可以来看看。

1、使用argparse组件，获取命令行参数；使用re组件，获取需要查找的字符串所在行

2、使用pandas组件，对文件进行排序。

3、命令行执行数据获取及排序，写入文件；

以下是完整代码：

#coding:utf-8
import re
import argparse
import pandas as pd
parser = argparse.ArgumentParser(description='manual to this script')
parser.add_argument('--ip', type=str, default = None)
parser.add_argument('--type', type=str, default=None)
args = parser.parse_args()
filterStr = args.ip + " " + args.type
f1=file('perf.csv','r')
perfdata=f1.readlines()
f1.close()
results = []
f2 = open('filter.csv', 'w')
f2.writelines(perfdata[0])
for i in perfdata:
    n = re.findall(filterStr, i)
    if n:
        f2.writelines(i)
f2.close()
df = pd.read_csv('filter.csv')
df = df.sort_values('elapsed',ascending = False)
df.to_csv('filterOrder.csv',index = False)

实例扩展：

Python对csv排序

#/usr/bin/evn python
# -*- coding: utf-8 -*-
import sys
from operator import itemgetter

# input_file = open(sys.argv[1])
input_file = open("D:\\tmp\\a.csv")
output_file = open("D:\\tmp\\asorted.csv","w")

table = []

for line in input_file:
  col = line.split('|') 
  col[0] = col[0].strip()
  col[1] = int(col[1])
  col[2] = int(col[2]) 
  col[3] = int(col[3].strip())
  table.append(col) #嵌套列表table[[8,8][*,*],...]

table_sorted = sorted(table, key=itemgetter(1,2),reverse=True)#先后按列索引1,2排序，降序排列

output_file.write('header' + '\n')
for row in table_sorted:          #遍历读取排序后的嵌套列表
  row = [str(x) for x in row]       #转换为字符串格式，好写入文本
  output_file.write("\t".join(row) + '\n')
  

input_file.close()
output_file.close()