这篇文章主要讲解了python对csv里的数据排序的代码详解,内容清晰明了,对此有兴趣的小伙伴可以学习一下,相信大家阅读完之后会有帮助。

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

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

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

以下是完整代码:

#coding:utf-8importreimportargparseimportpandasaspdparser=argparse.ArgumentParser(description='manualtothisscript')parser.add_argument('--ip',type=str,default=None)parser.add_argument('--type',type=str,default=None)args=parser.parse_args()filterStr=args.ip+""+args.typef1=file('perf.csv','r')perfdata=f1.readlines()f1.close()results=[]f2=open('filter.csv','w')f2.writelines(perfdata[0])foriinperfdata:n=re.findall(filterStr,i)ifn: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 sysfrom 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()

看完上述内容,是不是对python对csv里的数据排序的代码详解有进一步的了解,如果还想学习更多内容,欢迎关注亿速云行业资讯频道。