如何解决 Python 中的 KeyError: <variable>？

Question

您好，我正在尝试打开带有外部文件标头的简单 csv 文件：

得到下一个文件名为：name.csv，下一个内容：

Leo,Days,Ju
Tomas,Lee,Bruce
Max,Perez,Smith

如果我编码：

import csv
sep = ','
with open('name.csv') as csvfile:
     fieldnames = ['name', 'paterno', 'materno']
     reader = csv.DictReader(csvfile,fieldnames)
     for row in reader:
         list = (row['name'], \
                 row['materno'])
         print (sep.join(list))

想要的结果如下：

Leo,Ju
Tomas,Bruce
Max,Smith

但如果有一个额外的文件，其标题名为 hdr_name.txt 与：

['name', 'paterno', 'materno']

有了这个新代码：

import csv
sep = ','
fieldnames =  open('hdr_name.txt', 'r').read()
with open('name.csv') as csvfile:
    print(fieldnames)
    reader = csv.DictReader(csvfile,fieldnames)
    for row in reader:
        list = (row['name'], \
            row['materno'])
        print (sep.join(list))

得到结果：

Traceback (most recent call last):
  File "<stdin>", line 5, in <module>
KeyError: 'name'

但如果我在字段名中询问“名称”，是否存在！

>>> 'name' in fieldnames
True
>>> 

我做错了什么，从外部文件打开标题？

Answer 1

fieldnames 是一个如下所示的字符串：

"['name', 'paterno', 'materno']"

当然，成员资格测试将返回 true，但这并不意味着 fieldnames 是一个列表。请记住，file.read 返回一个字符串 - 您仍然需要将其转换为列表。

这看起来不像 JSON，所以我推荐ast：

import ast
with open('hdr_name.txt', 'r') as f:
     fieldnames = ast.literal_eval(f.read().strip())