python使用参数对嵌套字典进行取值的方法

时间:2022-09-18 09:34:47

因一些特殊需求需要以参数的形式获取字典中特定的值,网上搜了一下并没有特别好的实现(并没有太认真去找~),所以自己实现了一个,以供大家参考:) 。

话不多说,直接上代码:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
def dict_get(dic, locators, default=none):
 
'''
 
:param dic: 输入需要在其中取值的原始字典 <dict>
:param locators: 输入取值定位器, 如:['result', 'msg', '-1', 'status'] <list>
:param default: 进行取值中报错时所返回的默认值 (default: none)
:return: 返回根据参数locators找出的值
 
'''
 
if not isinstance(dic, dict) or not isinstance(locators, list):
 return default
 
value = none
 
for locator in locators:
 if not type(value) in [dict, list] and isinstance(locator, str) and not can_convert_to_int(locator):
 try:
  value = dic[locator]
 except keyerror:
  return default
 continue
 if isinstance(value, dict):
 try:
  value = dict_get(value, [locator])
 except keyerror:
  return default
 continue
 if isinstance(value, list) and can_convert_to_int(locator):
 try:
  value = value[int(locator)]
 except indexerror:
  return default
 continue
 
return value
 
def can_convert_to_int(input):
try:
 int(input)
 return true
except baseexception:
 return false

best practice

好的我们来进行一次简单的最佳实践:)

?
1
2
3
4
if __name__ == '__main__':
 dict_test = {"result": {"code": "110002", "msg": [{'status': 'ok'}, {'status': 'failed'}]}}
 result = dict_get(dict_test, ['result', 'msg', '-1', 'status'])
 print(result)

下面是控制台的输出,大家可以看到输出是符合预期结果的:)

failed

process finished with exit code 0

这次分享到此为止~ 我们有缘再见:)

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持服务器之家。

原文链接:https://juejin.im/post/5cbeb124f265da039d3284ad