现在的位置: 首页 > 编程语言 > 正文

飘逸的python – 实现一个pretty函数漂亮的输出嵌套字典

2019年06月08日 编程语言 ⁄ 共 979字 ⁄ 字号 评论关闭

示例:

d = { "root": { "folder2": { "item2": None, "item1": None }, "folder1": { "subfolder1": { "item2": None, "item1": None }, "subfolder2": { "item3": None } } } }

美观输出是:

观察一下特点:

1.同一level的key左对齐,即indent相同
2.在,号后换行
3.如果value是字典,即嵌套字典,则嵌套字典在下一个level,每一个level的key的indent都不一样

思路分析:
这是个"拼接字符串"问题,元素有""{}:,\n和空格indent.
遍历一个个(k, v)键值对,拼接yield起来,遇到嵌套字典就递归

即递归+yield

上代码。

#coding=utf-8
def pretty_dict(obj, indent=' '):
    def _pretty(obj, indent):
        for i, tup in enumerate(obj.items()):
            k, v = tup
            #如果是字符串则拼上""
            if isinstance(k, basestring): k = '"%s"'% k
            if isinstance(v, basestring): v = '"%s"'% v
            #如果是字典则递归
            if isinstance(v, dict):
                v = ''.join(_pretty(v, indent + ' '* len(str(k) + ': {')))#计算下一层的indent
            #case,根据(k,v)对在哪个位置确定拼接什么
            if i == 0:#开头,拼左花括号
                if len(obj) == 1:
                    yield '{%s: %s}'% (k, v)
                else:
                    yield '{%s: %s,\n'% (k, v)
            elif i == len(obj) - 1:#结尾,拼右花括号
                yield '%s%s: %s}'% (indent, k, v)
            else:#中间
                yield '%s%s: %s,\n'% (indent, k, v)
    print ''.join(_pretty(obj, indent))

d = { "root": { "folder2": { "item2": None, "item1": None }, "folder1": { "subfolder1": { "item2": None, "item1": None }, "subfolder2": { "item3": None } } } }
pretty_dict(d)

抱歉!评论已关闭.