在Python中扁平化未知深度的词典(等)列表的列表(噩梦般的JSON结构)

时间:2011-12-12 16:35:28

标签: python json list recursion dictionary

我正在处理一个JSON结构,它以这样的结构输出给我:

[{u'item': u'something',
  u'data': {
            u'other': u'',
            u'else':
               [
                  {
                    u'more': u'even more',
                    u'argh':
                         {
                            ...etc..etc

如您所见,这些是嵌套的词组和列表。 有很多关于递归地扁平这些的讨论,但是我还没有找到一个可以处理字典列表的字典,这些字典又可以包含列表字典,列表列表,字典字典等;深度不详!在某些情况下,深度可能高达100左右。 到目前为止我一直在尝试这个没有太多运气(python 2.7.2):

def flatten(structure):
    out = []
    for item in structure:
        if isinstance(item, (list, tuple)):
            out.extend(flatten(item))
        if isinstance(item, (dict)):
            for dictkey in item.keys():
                out.extend(flatten(item[dictkey]))
        else:
            out.append(item)
    return out

有什么想法吗?

更新 的 这非常有效:

def flatten(l):
    out = []
    if isinstance(l, (list, tuple)):
        for item in l:
            out.extend(flatten(item))
    elif isinstance(l, (dict)):
        for dictkey in l.keys():
            out.extend(flatten(l[dictkey]))
    elif isinstance(l, (str, int, unicode)):
        out.append(l)
    return out

1 个答案:

答案 0 :(得分:10)

由于数据的深度是任意的,因此更容易使用递归来展平它。此函数创建一个平面字典,其中每个数据项的路径都作为键组成,以避免冲突。

您可以稍后使用for key in sorted(dic_.keys())检索其内容。

我没有对它进行测试,因为您没有提供数据的“有效”摘要。

def flatten(structure, key="", path="", flattened=None):
    if flattened is None:
        flattened = {}
    if type(structure) not in(dict, list):
        flattened[((path + "_") if path else "") + key] = structure
    elif isinstance(structure, list):
        for i, item in enumerate(structure):
            flatten(item, "%d" % i, path + "_" + key, flattened)
    else:
        for new_key, value in structure.items():
            flatten(value, new_key, path + "_" + key, flattened)
    return flattened