正确放置 CSV 文件且 CSV 文件中存在密钥时引发 KeyError

Question

我有这个 Python 代码：

import csv

csvfile = 'stations-nl-2014-01.csv'

try:
    f = open(csvfile, 'r')
    reader = csv.DictReader(f, delimiter=';')

    for row in reader:
        print(row['name'], '--', row['type'])
finally:
    f.close()

我可以使用常规的 csv.reader() 方法，但有人告诉我使用 csv.DictReader() 是最佳做法，因为您可以按名称选择列，这使代码更具可读性。但是，当我使用普通的 csv.reader() 方法时，没有显示错误。当我使用 DictReader 时，我收到以下错误：

print(row['name'], '--', row['type']) KeyError: 'name'

当我将代码更改为：

import csv

csvfile = 'stations-nl-2014-01.csv'

try:
    f = open(csvfile, 'r')
    reader = csv.reader(f, delimiter=';')

    for row in reader:
        print(row)
finally:
    f.close()

...一切正常。但是使用 csv.DictReader() 并按列名选择行会引发错误。有人知道怎么修这个东西吗？我已经仔细检查了 .csv 文件的路径和列名，一切都正确（没有 DictReader 也可以正常工作）。

Answer 1

这是一个不好的问题，因为没有人知道 CSV 文件中的内容。

但是我遇到了同样的情况，在第一个键上我遇到了键错误。 事实证明，最初在 Excel 中创建的文件是用 UTF-8 编码保存的，并且有一个静默 '\ufeff' 前缀

我的解决方法是将每个 dict 转换为使用 ascii 编码的字符串

def convert_row( row ):
row_dict = {}
for key, value in row.items():
    keyAscii = key.encode('ascii', 'ignore' ).decode()
    valueAscii = value.encode('ascii','ignore').decode()
    row_dict[ keyAscii ] = valueAscii
return row_dict

这也可以通过 map / lambda 来完成，例如

def ascii_encode( str ):
    return str.encode( 'ascii', 'ignore' ).decode()

def convert_row( row ):
    return dict(map( lambda(key,value) :
        (ascii_encode(key),ascii_encode(value) ), row.items() ) )

或者类似的东西

我的回答中的主要问题是，如果文件是 utf-8，可能需要对字符串进行 ascii_encode。

Answer 2

1) 我在网上搜索了文件“stations-nl-2014-01.csv”，发现它没有“名称”列。虽然存在列“类型”。

它有以下列： 代码 uic naam middel_naam korte_naam rdt_url type geo_lat geo_lng

您确定不应该查找 row['naam']？

2) 如果上述方法不起作用，您可以使用 row.keys() 获取“行”字典包含的键列表。这将为您提供 csv.DictReader 实际识别的键的列表。