Python 3: Flattening nested dictionaries and lists within dictionaries

后端 未结 2 1436
一生所求
一生所求 2020-12-05 22:02

I am dealing with a complex nested dictionary and list data structure. I need to flatten the data and bring all nested items to level 0. See below example for more clarity :

2条回答
  •  情书的邮戳
    2020-12-05 22:58

    In my project, I am using an updated version of function from DSMs answer to flatten dict which may contain other dict or list or list of dict. I hope it will be helpful.

    def flatten(input_dict, separator='_', prefix=''):
        output_dict = {}
        for key, value in input_dict.items():
            if isinstance(value, dict) and value:
                deeper = flatten(value, separator, prefix+key+separator)
                output_dict.update({key2: val2 for key2, val2 in deeper.items()})
            elif isinstance(value, list) and value:
                for index, sublist in enumerate(value, start=1):
                    if isinstance(sublist, dict) and sublist:
                        deeper = flatten(sublist, separator, prefix+key+separator+str(index)+separator)
                        output_dict.update({key2: val2 for key2, val2 in deeper.items()})
                    else:
                        output_dict[prefix+key+separator+str(index)] = value
            else:
                output_dict[prefix+key] = value
        return output_dict
    

提交回复
热议问题