Python删除CSV文件特定行

删除第一列值为'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)




猜你喜欢

转载自blog.csdn.net/feelingwild/article/details/106968151