天天看点

python3通过字典的value(值)获取对应的key(键)

这个问题是在写接口自动化程序的过程中,有个场景希望能通过字典的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时再继续往下执行。