python教程—Python DictWriter编写UTF-8编码的CSV文件-Python实用宝典

python教程—Python DictWriter编写UTF-8编码的CSV文件

csv模块文档有一个将所有内容转换为UTF8的示例:它还有一个UnicodeWriter类。
  1. 我有一个包含unicode字符串的字典列表。
  2. csv。DictWriter可以将字典列表写入CSV文件。
  3. 我希望CSV文件用UTF8编码。
  4. csv模块无法处理将unicode字符串转换为UTF8。
  5. csv模块文档有一个将所有内容转换为UTF8的例子:

      def utf_8_encoder(unicode_csv_data): for line in unicode_csv_data: yield line.encode('utf-8')
  6. 它还有一个UnicodeWriter类。

但是…我如何使DictWriter与这些工作?难道他们不需要把自己插入到它的中间,来捕获被分解的字典,并在它将它们写入文件之前对它们进行编码吗?我不明白。

回答

<强>更新:第三方unicodecsv模块为您实现了这个7年前的答案。下面的代码示例。还有一个不需要第三方模块的python3解决方案

<强>原始Python 2答案

如果使用Python 2.7或更高版本,在传递给DictWriter之前,使用dict理解将字典重新映射到utf-8:

    # coding: utf-8 import csv D = {'name':u'马克','pinyin':u'mǎkè'} f = open('out.csv','wb') f.write(u'ufeff'.encode('utf8')) # BOM (optional...Excel needs it to open UTF-8 file properly) w = csv.DictWriter(f,sorted(D.keys())) w.writeheader() w.writerow({k:v.encode('utf8') for k,v in D.items()}) f.close()

你可以用这个想法来更新UnicodeWriter到DictUnicodeWriter:
utf - 8编码:
导入csv
进口cStringIO
进口编解码器

    class DictUnicodeWriter(object): def __init__(self, f, fieldnames, dialect=csv.excel, encoding="utf-8", **kwds): # Redirect output to a queue self.queue = cStringIO.StringIO() self.writer = csv.DictWriter(self.queue, fieldnames, dialect=dialect, **kwds) self.stream = f self.encoder = codecs.getincrementalencoder(encoding)() def writerow(self, D): self.writer.writerow({k:v.encode("utf-8") for k,v in D.items()}) # Fetch UTF-8 output from the queue ... data = self.queue.getvalue() data = data.decode("utf-8") # ... and reencode it into the target encoding data = self.encoder.encode(data) # write to the target stream self.stream.write(data) # empty queue self.queue.truncate(0) def writerows(self, rows): for D in rows: self.writerow(D) def writeheader(self): self.writer.writeheader() D1 = {'name':u'马克','pinyin':u'Mǎkè'} D2 = {'name':u'美国','pinyin':u'Měiguó'} f = open('out.csv','wb') f.write(u'ufeff'.encode('utf8')) # BOM (optional...Excel needs it to open UTF-8 file properly) w = DictUnicodeWriter(f,sorted(D.keys())) w.writeheader() w.writerows([D1,D2]) f.close()

<强>Python 2 unicodecsv示例:

    # coding: utf-8 import unicodecsv as csv D = {u'name':u'马克',u'pinyin':u'mǎkè'} with open('out.csv','wb') as f: w = csv.DictWriter(f,fieldnames=sorted(D.keys()),encoding='utf-8-sig') w.writeheader() w.writerow(D)

Python 3: <强> < / >强

此外,Python 3内置的csv模块本身就支持Unicode:

    # coding: utf-8 import csv D = {u'name':u'马克',u'pinyin':u'mǎkè'} # Use newline='' instead of 'wb' in Python 3. with open('out.csv','w',encoding='utf-8-sig',newline='') as f: w = csv.DictWriter(f,fieldnames=sorted(D.keys())) w.writeheader() w.writerow(D)

​Python实用宝典 (pythondict.com)
不只是一个宝典
欢迎关注公众号:Python实用宝典

本文由 Python实用宝典 作者:Python实用宝典 发表,其版权均为 Python实用宝典 所有,文章内容系作者个人观点,不代表 Python实用宝典 对观点赞同或支持。如需转载,请注明文章来源。
0

发表评论