这个问题是在写接口自动化程序的过程中,有个场景希望能通过字典的value获取对应的key,在网上找到了解决方案,这里自己加一些注解放出来,希望能让大家明白是如何实现的。
网友的原代码形式如下
def getDictKey_1(myDict, value):
return [k for k, v in myDict.items() if v == value]
代码特别简洁,返回一个列表,但是可能有些同学不太明白如何实现的,下面对该代码进行一个变装:
def getDictKey_2(myDict, value):
keyList = []
for k, v in myDict.items():
if v == value:
keyList.append(k)
return keyList
上面这两个代码的效果是一样的,测试结果如下
d = {'中国':'北京', '首都':'北京', '河南':'郑州', '余额': 0.01}
print("getDictKey_1(d, '北京')=", getDictKey_1(d, '北京'))
print("getDictKey_1(d, '郑州')=", getDictKey_1(d, '郑州'))
print("getDictKey_1(d, 0.01)=", getDictKey_1(d, 0.01))
print("getDictKey_1(d, '0.01')=", getDictKey_1(d, '0.01'))
print("getDictKey_2(d, '北京')=", getDictKey_2(d, '北京'))
print("getDictKey_2(d, '郑州')=", getDictKey_2(d, '郑州'))
print("getDictKey_2(d, 0.01)=", getDictKey_2(d, 0.01))
print("getDictKey_2(d, '0.01')=", getDictKey_2(d, '0.01'))
执行结果:
getDictKey_1(d, '北京')= ['中国', '首都']
getDictKey_1(d, '郑州')= ['河南']
getDictKey_1(d, 0.01)= ['余额']
getDictKey_1(d, '0.01')= []
getDictKey_2(d, '北京')= ['中国', '首都']
getDictKey_2(d, '郑州')= ['河南']
getDictKey_2(d, 0.01)= ['余额']
getDictKey_2(d, '0.01')= []
那么,为什么这样就可以获取到key呢?重点就在于字典的items()这个方法的返回值上,见下面的代码
d = {'中国': '北京', '首都': '北京', '河南': '郑州', '余额': 0.01}
print(d.items())
结果
dict_items([('中国', '北京'), ('首都', '北京'), ('河南', '郑州'), ('余额', 0.01)])
可以看到,items()方法会遍历字典的每个key-value键值对并将每个键值对以元组的形式保存,其中key为第一个元素,value为第二个元素,即(key, value),并且会将所有的键值对元组以列表的形式返回。因此可以对items()返回值的列表进行遍历后处理(key, value)就可以得到想要的key。需要注意的是,一个value可能会对应多个key。
容错处理
上面两个函数在正常传参的情况下可以正确返回想要的结果,但是如果第一个参数myDict不是字典形式,就会抛出异常,如下:
d = "{'中国':'北京', '首都':'北京', '河南':'郑州', '余额': 0.01}"
print("getDictKey_1(d, '北京')=", getDictKey_1(d, '北京'))
结果:
Traceback (most recent call last):
File "E:/feedback_record/test.py", line 131, in <module>
print("getDictKey_1(d, '北京')=", getDictKey_1(d, '北京'))
File "E:/feedback_record/test.py", line 99, in getDictKey_1
return [k for k, v in myDict.items() if v == value]
AttributeError: 'str' object has no attribute 'items'
很好处理,做个判断就可以了,如下:
def getDictKey_3(myDict, value):
if not isinstance(myDict, dict):
return False
keyList = []
for k, v in myDict.items():
if v == value:
keyList.append(k)
return keyList
d = "{'中国':'北京', '首都':'北京', '河南':'郑州', '余额': 0.01}"
print("getDictKey_3(d, '北京')=", getDictKey_3(d, '北京'))
结果:
getDictKey_3(d, '北京')= False
注意:
大家正式使用的时候,要做个判断,正确返回了key时再继续往下执行。