您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

使用CSV文件中的Python删除行

使用CSV文件中的Python删除行

你很亲密 目前,您将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
python 2022/1/1 18:25:06 有176人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶