將嵌套字典轉換為鍵值

Question

我有一個復雜的字典，像這樣：

d = {
  "key1": "value1",
  "key2": {
    "key3": "value3",
    "key4": {
      "key5": "value5",
      "key6": [{"keys1": "value_keys1"}]
    }
}

我想打開一個簡單的鍵值

{
  "key1": "value1",
  "key2_key3": "value3",
  "key2_key4_key5": "value5",
  "key2_key4_key6_0_keys1": "value_keys1",

}

我怎么能在python上做到這一點？

我試過這個：

def flat_dict_helper(prepend, d):
    if len(prepend) > 0:
        prepend = prepend + "_"
    for k in d:
        i = d[k]
        if isinstance(i, dict):
            r = flat_dict_helper(prepend + k, i)
            for j in r:
                yield j
        else:
            yield (prepend + k, i)


def flat_dict(d):
    return dict(flat_dict_helper("", d))

但這在某些情況下不起作用，例如當里面有一個列表時

Answer 1

您必須在列表的情況下指定行為：

def flat_dict_helper(prepend, d):
    prepend += "_" if len(prepend) else prepend
    for k in d:
        i = d[k]
        if isinstance(i, dict):
            r = flat_dict_helper(prepend + k, i)
            for j in r:
                yield j
        elif isinstance(i, list):
            for idx, element in enumerate(i):
                r = flat_dict_helper('{}{}_{}'.format(prepend, k, idx), element)
                for j in r:
                    yield j
        else:
            yield (prepend + k, i)



def flat_dict(d):
    return dict(flat_dict_helper("", d))

>>> flat_dict(d)
{'key1': 'value1', 'key2_key3': 'value3', 'key2_key4_key5': 'value5', 'key2_key4_key6_0_keys1': 'value_keys1'}