python封装获取字典值函数,支持多维字典

  • 2023-01-03
    • 可自定义键名的分割符,默认为【.】

前些时间写了个php获取数组的。

刚好现在python的项目也需要同样的功能,就也写了一个,算法都一样,就不多说别的了,直接上代码。

# coding=UTF-8
def getval(dictdata, key, default=None, separator='.'):
    if not isinstance(dictdata, dict):
        return default
    keys = key if isinstance(key, list) else key.split(separator)
    firstkey = keys[0]
    if len(keys) == 1:
        # return dictdata[firstkey] if firstkey in dictdata else default
        if firstkey in dictdata:
            return dictdata[firstkey]
        else:
            return default
    else:
        if firstkey not in dictdata:
            return default
        dictdata = dictdata[firstkey]
        keys.remove(firstkey)
        return getval(dictdata, keys, default, separator)

调用方法跟php版的一样

data = {
    'site': {
        'name': '虚拟笔记',
        'domain': 'www.notevm.com',
        'page': {
            'tags': {
                'python': 'https://www.notevm.com/a/special/python'
            }
        }
    }
}

name = getval(data, 'site.name')
print('获取二维字典:%s' % name)

python = getval(data, 'site.page.tags.python')
print('获取四维字典:%s' % python)

url = getval(data, 'site.url', 'https://www.notevm.com/')
print('元素不存在返回默认值:%s' % url)
python获取字典的值

自定义分割符

如果数组的某个键名中包含【.】的话,可以自定义传入键名的分割符

data = {
    'site': {
        'name': '虚拟笔记',
        'domain': 'www.notevm.com',
        'page': {
            'tags.python': 'https://www.notevm.com/a/special/python'
        }
    }
}
python = getval(dictdata=data, key='site|page|tags.python', separator='|')
print(python)

原创文章,作者:小哆啦,如若转载,请注明出处:https://www.notevm.com/a/6083.html

(0)
上一篇 2022年6月7日 10:30
下一篇 2022年6月8日 14:22

相关推荐

发表回复

登录后才能评论