python - 如何使用 csv 模块为 "delimiter"提供 2 个字符？

Question

我正在尝试生成带有分隔符“@|@”的 csv，但是我无法通过以下代码实现。

import csv
ifile  = open('test.csv', "rb")
reader = csv.reader(ifile)
ofile  = open('ttest.csv', "wb")
writer = csv.writer(ofile, delimiter='|', quotechar='"', quoting=csv.QUOTE_ALL)
for row in reader:
    writer.writerow(row)
ifile.close()
ofile.close()

运行时，抛出以下错误。

Traceback (most recent call last):

File "csvfile.py", line 6, in <module>
writer = csv.writer(ofile, delimiter='@|@', quotechar='"', quoting=csv.QUOTE_ALL)
TypeError: "delimiter" must be an 1-character string

我怎样才能做到这一点？

Answer 1

在 csv 中 documentation他们说

A one-character string used to separate fields. It defaults to ','.

因此您可以将此作为替代方案。

csv.reader((line.replace('@|@', '|') for line in ifile), delimiter='|')

所以完整的代码是，

import csv
ifile  = open('test.csv', "rb")
reader = csv.reader((line.replace('@|@', '|') for line in ifile), delimiter='|')
ofile  = open('ttest.csv', "wb")
writer = csv.writer(ofile, delimiter='|', quotechar='"', quoting=csv.QUOTE_ALL)
for row in reader:
    writer.writerow(row)
ifile.close()
ofile.close()