如何在列表理解中解包列表

Question

我有一个字典列表，每个字典都可以嵌套一个字典列表，如下所示：

mydictlist = [{'name':'foo'}, {'name':'bar','next-level':[{'name':'next-level-foo'}, {'name':'next-level-bar'}] } ]

我试图在列表理解中展平某种属性，如名称，如下所示：

flattened = [ *['{}_{}'.format(iter['name'],next['name']) for next in iter] if 'next-level' in iter else '{}'.format(iter['name']) for iter in mydictlist]

得到类似的东西：

['foo', 'bar_next-level-foo', 'bar_next-level-bar']

但这会导致错误。？ 我可以使用 for 和 if 没有列表理解（并且已经做到了）但我想知道使用列表（或元组解包）和列表理解的正确语法是什么？

Answer 1

mydictlist = [{'name':'foo'}, {'name':'bar','next-level':[{'name':'next-level-foo'}, {'name':'next-level-bar'}] } ]

def flatten(currlist, prefix = None, delimiter = "_"):
    l = []
    for x in currlist:
        print (l)
        if "next-level" in x:
            if prefix:
                prefix = delimiter.join([prefix, x["name"]])
            else:
                prefix = x["name"]
            c = flatten(currlist=x["next-level"], prefix=prefix, delimiter=delimiter)
            l.extend(c)
        else:
            if not prefix:
                item = x["name"]
            else:
                item = delimiter.join([prefix, x["name"]])
            l.append(item)
    return l

flatten(currlist=mydictlist)

Answer 2

这里是你想要的差不多

result = [x['name'] if "next-level" not in x else [ j['name'] for j in x['next-level']] for x in mydictlist ]
print(result)
# output: ['foo', ['next-level-foo', 'next-level-bar']]
result2 = []
def flettern_list(lst):
    for s in lst:
        if isinstance(s, str):
            result2.append(s)
        elif isinstance(s, list):
            flettern_list(s)
flettern_list(result)
print(result2) #output: ['foo', 'next-level-foo', 'next-level-bar']

Answer 3

这是一个相当通用的解决方案，它将在任何类型的 dict 和 list 组合中递归。 这将返回名称值的平面列表。

def get_leafs_as_list(obj, key):
    l = []
    if isinstance(obj, list):
        for v in obj:
            l.extend(get_leafs_as_list(v, key))
    elif isinstance(obj, dict):
        for k, v in obj.items():
            if k == key:
                l.append(v)
            else:
                l.extend(get_leafs_as_list(v, key))
    return l

mydictlist = [{'name':'foo'}, {'name':'bar','next-level':[{'name':'next-level-foo'}, {'name':'next-level-bar'}] } ]

print(get_leafs_as_list(mydictlist, 'name'))

这将返回：

['foo', 'bar', 'next-level-foo', 'next-level-bar']

Answer 4

我能做的最接近的是在必要时提供“假”第二级，然后忽略它：

mydictlist = [{'name':'foo'}, {'name':'bar','next-level':[{'name':'next-level-foo'}, {'name':'next-level-bar'}] } ]

flattened = [ '{}_{}'.format(item['name'],next['name']) if next['name'] else item['name'] for item in mydictlist for next in (item['next-level'] if 'next-level' in item else  [{'name':None}]) ]

Output：

 ['foo', 'bar_next-level-foo', 'bar_next-level-bar']

在列表理解中似乎没有条件循环的方法。 因此，如果有两个for -s，它们都将运行，但可以有条件地生成第二个循环的“主题”（最初我使用 lambda 执行此操作，但显然带括号的表达式就足够了）。

您的尝试有一个错误， for next in iter应该是for next in iter['next-level'] （另外， iter()是一个内置的 function，所以我将它重命名为item ）。
如果你尝试了一些不那么雄心勃勃的事情，你会遇到一个明确的错误信息： iterable unpacking cannot be used in comprehension 。