给定任意dict数据，请实现方法提取数据字典，即将树状多级字段压缩为一级字段

Question

import json

def solution(data: dict):
    # 在这⾥写代码，可自行添加更多方法
    pass


def main():
    # 测试数据1 - 较简单，方便测试
    test1 = {
        "a": 1,
        "b": {
            "c": "test"
        }
    }

    # 期望结果1
    result1 = {
        "a": "int",
        "b": "dict",
        "b.c": "str"
    }

    # 提取数据字典并校验结果
    ret1 = solution(test1)
    print('ret1 =', json.dumps(ret1, indent=4))
    assert ret1 == result1, 'test1结果不正确'

    # 测试数据2 - 略复杂，有数组嵌套
    # 假定数组内元素类型是相同的，不考虑不相同的情况
    test2 = {
        "x": {
            "x1": "aa",
            "x2": [11, 22],  # 注意元素为基本类型
            "x3": {
                "m": {
                    'n': 123
                },
            }
        },
        "y": [  # 注意元素为复杂类型
            {
                "p": 1,
                "q": 'aa',
            },
            {
                "p": 3,
                "q": "bb"
            }
        ],
        "z": True
    }

    # 期望结果2
    result2 = {
        "x": "dict",
        "x.x1": "str",
        "x.x2": "list",
        "x.x3": "dict",
        "x.x3.m": "dict",
        "x.x3.m.n": "int",
        "y": "list",
        "y.p": "int",
        "y.q": "str",
        "z": "bool"
    }

    # 提取数据字典并校验结果
    ret2 = solution(test2)
    print('ret2 =', json.dumps(ret2, indent=4))
    assert ret2 == result2, 'test2结果不正确'

if __name__ == "__main__":
    main()

Answer 1

各种数据类型都测试了，没问题的话请点击采纳