提取csv文件中需要的部分

 1 # coding=utf-8
 2 import csv,sys,datetime
 3 
 4 time1 = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
 5 # input_file = sys.argv[1]
 6 input_file = r"/Users/vito/Desktop/SW/Python/python_study/csv/records.csv"
 7 
 8 output_file = input_file+"_Fail_Item_"+time1+".csv"
 9 # input_file = r"/Users/vito/Desktop/s/records.csv"
10 # output_file = r"/Users/vito/Desktop/s/records_Fail_Item.csv"
11 
12 with open(input_file, 'r', newline='') as csv_in_file:
13     with open(output_file, 'a', newline='') as csv_out_file:
14         filereader = csv.reader(csv_in_file)
15         filewriter = csv.writer(csv_out_file)
16         header = next(filereader)
17         filewriter.writerow(header)
18         for i,rows in enumerate(filereader):
19             # if i == 0:
20             #     second = rows
21             #     print(rows)
22             #     filewriter.writerow(second)
23             # else:
24                 status = str(rows[12]).strip()
25                 if status == 'FAIL' or rows[0] != "":
26                     filewriter.writerow(rows)
27                     print("正在处理第{0}行数据...".format(i))