Python中如何用列表推导式正确编写代码计算result变量的值

你编写的伪代码问题在于使用了对象属性访问符.，但你给出的示例代码中所有变量均为列表嵌套结构，不存在pages、boxes类属性，需要替换为索引访问。

首先明确示例代码的层级关系：

• npr[1] 对应你伪代码中的npr.pages，是存储所有page的列表
• npr[1][i] 对应你伪代码中的npr.pages[i]，代表第i个page，每个page包含多个boxes集合
• 每个boxes集合中存储多个box元素

根据不同的需求，有两种常见写法： #

场景1：取第i个page中指定第j个boxes集合的所有box #

写法如下：

result = [box for box in npr[1][i][j]]

示例：取第0个page的第0个boxes集合的所有box，运行后得到的result为[[1,2], [1,2], [1,2]] #

场景2：取第i个page中所有boxes集合的全部box（合并结果） #

如果需要将第i个page内所有boxes里的box合并到同一个列表中，使用两层循环的列表推导式：

result = [box for boxes in npr[1][i] for box in boxes]

示例：取第0个page的所有box，运行后得到的result为[[1,2], [1,2], [1,2], [1,2], [1,2], [1,2]] #

补充说明 #

如果你实际业务场景中npr是自定义类的实例，本身存在pages属性，且每个page对象存在boxes属性，那么你原本编写的伪代码result = [box for box in npr.pages[i].boxes]本身是合法可用的。

内容的提问来源于stack exchange，提问作者shantanu rahut