删除第一列值为'1'的行
# coding=gbk
import csv
import os
file_old = 'data/test_1.csv'
file_temp = 'data/test_temp.csv'
with open(file_old, 'r', newline='', encoding='gbk') as f_old, \
open(file_temp, 'w', newline='', encoding='gbk') as f_temp:
f_csv_old = csv.reader(f_old)
f_csv_temp = csv.writer(f_temp)
for i, rows in enumerate(f_csv_old): # 保留header
if i == 0:
f_csv_temp.writerow(rows)
break
for rows in f_csv_old:
if rows[0] != '1': # 删除第一列值为1的行
f_csv_temp.writerow(rows)
os.remove(file_old)
os.rename(file_temp, file_old)