你非常接近;目前你将行[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)
请注意,输入是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