欢迎来到Python教程自学网!
当前位置: 首页 > Python基础教程 > 如何用python给csv里的数据排序

如何用python给csv里的数据排序

   

用python给csv里的数据排序的方法:1、使用argparse组件,获取命令行参数及数据;2、使用re组件,获取需要查找的字符串所在行;3、使用pandas组件对文件进行数据排序;4、获取执行后的文件数据即可。

如何用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)