你很亲密 目前,您将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