CSV 模块的编写器不允许我写出二进制文件

Question

我尝试在打开文件时仅使用“w”标签，但它使行间距加倍，导致读取无法正常工作。所以我发现更改为“wb”将是正确的格式。现在我使用“wb”标志，我无法让 csv.writer.writerow() 工作。我已经对所有字符串进行了编码，但不明白为什么我不断收到此错误。我看到的所有问题都说 b'string here' 或 myString.encode('ascii') 解决了我得到的错误，但它并没有为我解决它。这是我所拥有的：

    dataWriter = csv.writer(open(fileName, 'wb'))
    for i in range(self.ui.table.rowCount()):
        rowData = [self.ui.table.item(i,0).text().encode('utf-8')\
        ,self.ui.table.item(i,1).text().encode('utf-8')\
        ,self.ui.table.item(i,2).text().encode('utf-8')\
        ,self.ui.table.item(i,3).text().encode('utf-8')\
        ,self.ui.table.item(i,4).text().encode('utf-8')]
        dataWriter.writerow(rowData)

我认为它可以工作，但它仍然给我以下错误： “类型错误：必须是字节或缓冲区，而不是 str” 在“dataWriter.writerow(rowData)”行上。

任何帮助将不胜感激。 谢谢。

Answer 1

您似乎正在运行 Python 3.x。 关于对 csv 文件使用二进制模式的建议适用于 Python 2.x。3.x 不需要编解码器模块 - 只需在打开时使用 encoding=whatever文件。 3.x 需要的是使用 newline='' 打开文件。这适用于阅读和写作，尽管没有书面记录（错误报告已提交）。解决双倍间距问题后，这将起作用：

import csv
data = [
    ['\xfforick', 123.456],
    ['polonius', 987.564],
    ]
with open('demo.csv', 'w', newline='', encoding='utf8') as f:
    writer = csv.writer(f)
    for row in data:
        writer.writerow(row)

输出文件的内容：

>>> open('demo.csv', 'rb').read()
b'\xc3\xbforick,123.456\r\npolonius,987.564\r\n'
>>>

建议：考虑代码的易读性......而不是

for i in range(self.ui.table.rowCount()):
    rowData = [self.ui.table.item(i,0).text().encode('utf-8')\
    ,self.ui.table.item(i,1).text().encode('utf-8')\
    ,self.ui.table.item(i,2).text().encode('utf-8')\
    ,self.ui.table.item(i,3).text().encode('utf-8')\
    ,self.ui.table.item(i,4).text().encode('utf-8')]
    dataWriter.writerow(rowData)

尝试

table = self.ui.table
for i in range(table.rowCount()):
    row = [table.item(i, j).text() for j in range(5)]
    writer.writerow(row)

Answer 2

在 Python 3 中，在二进制模式下使用 open 会创建一个 io.BufferedWriter，它需要字节，而不是字符串。通过使用 encode 方法，您可以将字符串转换为字节；但我认为 cvs.writer.writerow 在写入之前将这些字节转换回字符串。

您应该尝试找出导致双倍行距的原因，而不是以二进制模式打开文件。我有两个问题：

1. 您使用什么平台？

2. print repr(self.ui.table.item(i,4).text()) 的输出是什么？

我的猜测是，brandizzi 的 strip() 方法可以工作，但如果不行，我们需要进行一些故障排除。

编辑：好的，John Machin 的帖子澄清了一切。在 Python 3 中解决此问题的正确方法是使用 newline='' 打开文件，这会禁用自动换行符翻译。 此错误报告包含一些有用的信息。

Answer 3

也许您可以让 codecs 模块为您执行 Unicode 编码，然后尝试这样的操作：

import codecs, csv

with codecs.open(fileName, 'w', encoding = 'utf_8') as f:
    writer = csv.writer(f)
    writer.writerow(['some string', 'some other string'])

Answer 4

我在尝试将 csv 写入 paramiko sftp 文件流时遇到了这个问题，该文件无法在文本模式下打开。

我想出的也适用于OP的解决方案是：

import codecs, csv, io
buf = io.BytesIO()
StreamWriter = codecs.getwriter( 'utf-8' )
w = csv.writer( StreamWriter( buf ))
w.writerow([ 'foo', 'bar' ])
w.writerow([ 'bar', 'foo' ])
print( repr( buf.getvalue() ))

产生正确的输出：

b'foo,bar\r\nbar,foo\r\n'

Answer 5

我一点也不感到惊讶。如果您要写出一个值为 13 的字节，模块如何判断这是二进制字段的一部分，还是 CSV 中新记录的开始？ CSV 文件不适合存储二进制数据。

如果你绝对需要它在那里，你可以研究 BASE 64 编码......

马丁