python - 为 csv.reader 预处理一行

Question

<分区>

Answer 1

csv.reader将愉快地从任何可迭代对象中读取行，包括生成器，因此我们可以使用生成器来进行引号转义。但我们还需要将我们的 escapechar 告诉 csv.reader .

我已将一些测试数据作为字符串列表嵌入到此代码中，但您可以将打开的文件传递给 fixlines 生成器。

import csv 

data = '''\
id,name,score
1,"something" like this",20
2,"another thing" like that",30
'''.splitlines()

def fixlines(lines):
    for row in lines:
        try:
            first = row.index('"') + 1
            last = row.rindex('"')
        except ValueError:
            pass
        else:
            stuff = row[first:last].replace('"', '\\"')
            row = row[:first] + stuff + row[last:]
        yield row

reader = csv.reader(fixlines(data), escapechar='\\')
for row in reader:
    print(row)

输出

['id', 'name', 'score']
['1', 'something" like this', '20']
['2', 'another thing" like that', '30']

此代码不会更改不包含双引号(如标题行)或一对双引号的行，但如果找到包含单个双引号的行，它可能不会完全按照您的要求执行 -引用。解决这个问题留给读者作为练习。 ;)

但是，它确实可以正确处理引号字符串中的逗号。

---

这也适用于 csv.DictReader :

reader = csv.DictReader(fixlines(data), escapechar='\\')
for row in reader:
    print(row)

输出

OrderedDict([('id', '1'), ('name', 'something" like this'), ('score', '20')])
OrderedDict([('id', '2'), ('name', 'another thing" like that'), ('score', '30')])