我收到了keyerror,无法从csv文件导入数据

时间:2018-03-09 09:47:47

标签: python

def load_data(label_dict):
    x = []
    y = []
    x_keys = ['FEAT_1','FEAT_2','FEAT_3','FEAT_4','FEAT_5','FEAT_6']
    y_keys = ['LABEL'];
    for item in label_dict:
        x.append([float(item[k]) for k in x_keys])
        y.append([float(item[k]) for k in y_keys])

    return (x,y)

错误是:

File "NID_1.py", line 54, in load_data
    x.append([float(item[k]) for k in x_keys])
KeyError: 'FEAT_1'

1 个答案:

答案 0 :(得分:0)

label_dict似乎是您迭代的字典列表。

该错误表示在扫描中遇到了x_keys中没有其中一个键的字典。

这里的答案实际上取决于您作为label_dict传入的内容,但是如果您想确保没有生成错误,您可以向字典item本身询问其密钥是,没有在x_keys列表

中对它们进行硬编码

Python2:

x.append([float(item[k]) for k in item.keys()])

Python 3:

x.append([float(item[k]) for k in list(item.keys())])