使用CSV文件中的Python删除行

Question

我只想删除第三行中值为‘0’的行。数据示例如下：

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()

任何帮助都会很棒

Answer 1

你很亲密
目前，您将row[2]整数与0进行比较，然后将其与字符串进行比较"0"。当您从文件中读取数据时，它是一个字符串而不是整数，因此这就是您的整数检查当前失败的原因：

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)

请注意，这input是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