python-Key Error问题

【问题描述】

当从数据库以标签形式大批量处理数据时，会因为不同的数据返回的标签数目不同而导致在使用标签来来索引数据时出现问题。（pandas转为数据框之后）

例如

数据：

a = {"id":[1,2,3,4,5],"birthday":['2000-01-01','',None,'2000-01-19',NaN],"name":['王菲','莫文蔚',NaN,None,'林宥嘉'],"score":[100,99,NaN,98,None]}
a = pd.DataFrame(test)
a

a['gender']

报错信息如下：

原因就是原数据中没有'gender'这个键。

【解决办法】

1) 设置条件语句:

index_of_a = list(a.columns.values)   #获取dataframe中包含字段的列表
b = 'gender'
c = 'name'
if b in list(index_of_a):
    print(a[b])
else:
    print('not exist!')
if c in list(index_of_a):
    print(a[c])
else:
    print('not exist!')

2）运用get():利用dict内置的get(key,default)方法，如果key存在，则返回其value,否则返回default;使用这个方法永远不会触发Key Error.

print(a.get('gender'))
print(a.get('gender','不存在！'))
print(a.get('name','不存在！'))

参考：python：操作dict时避免出现Key Error的几种方法

（仅供个人学习，不负责任~~~~~~~~~~~~）