Python读取csv-BOM嵌入到第一个密钥

Question

我正在使用Python 2.7.12。通过这个代码片段，我保存了一个utf-8CSV文件。我在文件的开头写下了BOM（字节顺序标记）。

import codecs
import csv

outputFile = open("test.csv", "wb")
outputFile.write(codecs.BOM_UTF8)
fieldnames = ["a", "b"]
writer = csv.DictWriter(outputFile, fieldnames, delimiter=";")
writer.writeheader()
row = dict([])
for i in range(10):
    row["a"] = str(i).encode("utf-8")
    row["b"] = str(i*2).encode("utf-8")
    writer.writerow(row)
outputFile.close()

我想加载csv文件：

import codecs
import csv
inputFile = open("test.csv", "rb")
reader = csv.DictReader(inputFile, delimiter=";")
for row in reader:
    print row["a"]
inputFile.close()

上面的代码将失败：KeyError:'a'如果我打印行键，它们的外观就是这样：[u'\ufeffa'，u'b']。BOM表已嵌入到键a中。我做错了什么？

Answer 1

在Python 3中，内置的打开函数是io.open的别名。

您只需要打开一个编码为UTF-8的BOM文件：

open(path, newline='', encoding='utf-8-sig')

import csv

...

with open(path, newline='', encoding='utf-8-sig') as csv_file:
    reader = csv.DictReader(csv_file, dialect='excel')
    for row in reader:
        print(row['first_name'], row['last_name'])

Answer 2

你必须告诉Open这是带有BOM的UTF-8。我知道这对io.open有用：

import io

.
.
.
inputFile = io.open("test.csv", "r", encoding='utf-8-sig')
.
.
.

您必须以文本模式打开文件，“r”而不是“rb”。