23

文字列をcsvファイルオブジェクトに書き込むために、python 2/3互換コードを書き込もうとしています。このコード:

line_as_list = [line.encode() for line in line_as_list]
writer_file =  io.BytesIO()
writer = csv.writer(writer_file, dialect=dialect, delimiter=self.delimiter)
for line in line_as_list:
    assert isinstance(line,bytes)
    writer.writerow(line)

Python3 で次のエラーが発生します。

>           writer.writerow(line)
E           TypeError: a bytes-like object is required, not 'str'

しかし、assert は型に問題がないのに、なぜcsvエラーが発生するのでしょうか。

BytesIOPython 2 と 3 の両方にしか使えないのですか? ここで問題はどこにありますか？

4

1 に答える 1

34

Python3 ではcsv.writer、テキスト モードで開かれたファイルのようなオブジェクトが必要です。Python2 ではcsv.writer、バイナリ モードで開かれたファイルのようなオブジェクトが必要です。

したがって、Python3 では を使用しio.StringIO、Python2 では を使用しますio.BytesIO。

import io
import csv
import sys
PY3 = sys.version_info[0] == 3

line_as_list = [u'foo', u'bar']
encoding = 'utf-8'

if PY3:
    writer_file =  io.StringIO()
else:
    writer_file =  io.BytesIO()
    line_as_list = [line.encode(encoding) for line in line_as_list]

writer = csv.writer(writer_file, dialect='excel', delimiter=',')
writer.writerow(line_as_list)
content = writer_file.getvalue()

if PY3:
    content = content.encode(encoding)

print(type(content))
print(repr(content))

Python3 では、上記のコードが出力されます

<class 'bytes'>
b'foo,bar\r\n'

Python2 では、上記のコードが出力されます

<type 'str'>
'foo,bar\r\n'
于 2016-06-22T17:34:43.577 に答える