带有 if 语句的嵌套列表推导

Question

我正在努力理解嵌套列表的理解，并阅读了出色的解释 here。

我正在翻译的问题是我的内部循环中有一个if 子句，我看不到如何将它应用于func() 步骤，因为我失去了我从中得到的计数器enumerate() 当我从嵌套循环转到列表理解时。

nested_list = [[{'a': 1, 'b': 2}, {'c': 3, 'd': 4}], [{'a': 5, 'b': 6}, {'c': 7, 'd': 8}]]
new_list = []
for c, x in enumerate(nested_list):
    for d, y in enumerate(x):
        if d == 1:
            new_list.append(y)
print(new_list)
[{'c': 3, 'd': 4}, {'c': 7, 'd': 8}]

嵌套列表理解可能类似于

new_list = [if ??? y
               for x in nested_list
                   for y in x]

...但我无法看到/思考如何获取该子句，因为我在嵌套列表理解下没有计数器。

有没有办法实现这一点，还是我应该坚持使用嵌套循环方法？

Answer 1

你可以改写如下：

new_list = [y for x in nested_list for d, y in enumerate(x) if d == 1]

循环是“自然”的顺序，条件是最后。仅当 d==1 时才包含 y

一个可能的结果（因为字典没有排序）：

[{'c': 3, 'd': 4}, {'c': 7, 'd': 8}]

请注意，在您的情况下，写起来更简单、更高效（O(n) 与 O(n**2)）：

new_list = [x[1] for x in nested_list]

唯一的区别是如果x 太短，后面的代码会中断，所以可能是测试长度：

new_list = [x[1] for x in nested_list if len(x)>1]

Answer 2

当我从嵌套循环转到列表理解时，我失去了从 enumerate() 获得的计数器。

那你为什么放弃它？列表推导式仍然可以使用与 for 循环相同的所有迭代器。

我的内部循环中有一个 if 子句，我看不到如何应用它

只要把它放在同一个地方。对于 for 循环和 if 过滤器，列表推导遵循相同的嵌套顺序；您的循环可以通过删除: 冒号直接映射，并将.append(...) 调用中的部分移到前面：

new_list = [y   # from new_list.append(y)
    for c, x in enumerate(nested_list)   # removed the :
        for d, y in enumerate(x)         # removed the :
            if d == 1]                   # removed the :

但是，您可以删除第一个 enumerate()，因为您不使用 c。您也可以删除第二个循环和enumerate()，因为您可以使用 indexing 代替：

new_list = [x[1] for x in nested_list]

您不需要在这里循环遍历x 中的所有值，这不是获取该值的一种非常有效的方法。

如果您的问题是 x 列表可能较短并且在索引 1 处没有元素，请先按长度过滤这些嵌套列表：

new_list = [x[1] for x in nested_list if len(x) > 1]

Answer 3

你想要列表中每个数组的第二个元素，所以你可以这样做：

new_list = [nested_list[i][1] for i in range(len(nested_list))]

更好

new_list  = [x[1] for x in nested_list]

Answer 4

print ([ y for c, x in enumerate(nested_list) for d, y in enumerate(x) if d == 1])