在CSV文件中使用Python删除行

6.5, 5.4, 0, 320
6.5, 5.4, 1, 320

因此需要删除第一行而第二行将保留.

我到目前为止的内容如下：

import csv
input = open('first.csv', 'rb')
output = open('first_edit.csv', 'wb')
writer = csv.writer(output)
for row in csv.reader(input):
    if row[2]!=0:
        writer.writerow(row)
input.close()
output.close()

任何帮助都会很棒

row[2]!="0":

此外,您可以使用with关键字使当前代码稍微更加pythonic,以便减少代码中的行,并且可以省略.close语句：

import csv
with open('first.csv', 'rb') as inp, open('first_edit.csv', 'wb') as out:
    writer = csv.writer(out)
    for row in csv.reader(inp):
        if row[2] != "0":
            writer.writerow(row)

请注意,输入是Python内置的,所以我使用了另一个变量名.

编辑：csv文件行中的值以逗号和空格分隔;在正常的csv中,它们只是逗号分隔,并且对“0”的检查可以工作,因此你可以使用strip(row [2])！= 0,或检查“0”.

更好的解决方案是更正csv格式,但是如果您想要保留当前格式,以下内容将适用于您给定的csv文件格式：

$cat test.py 
import csv
with open('first.csv', 'rb') as inp, open('first_edit.csv', 'wb') as out:
    writer = csv.writer(out)
    for row in csv.reader(inp):
        if row[2] != " 0":
            writer.writerow(row)
$cat first.csv 
6.5, 5.4, 0, 320
6.5, 5.4, 1, 320
$python test.py 
$cat first_edit.csv 
6.5, 5.4, 1, 320