Python三维数组拆分问题：如何用列表推导式保持多维结构？

你最初的列表推导式问题在于使用了两层并列的for循环，这会直接扁平化原有的嵌套结构。要保留三维数组的层级，你需要嵌套列表推导式，让它和输入数组的深度一一对应。

正确的列表推导式写法 #

生成y（取每个内层子列表的前3个元素）：

y = [[j[:3] for j in i] for i in x]

生成z（取每个内层子列表的后2个元素）：

z = [[j[3:] for j in i] for i in x]

为什么这样可行？ #

以y为例拆解逻辑：

1. 外层推导式[...] for i in x遍历x中的每一个顶层元素（也就是每一组子列表）。
2. 对于每一组i，内层推导式[j[:3] for j in i]会处理组内的每一个子列表j，截取它的前3个元素。
3. 这种嵌套结构保证了x中的每一个顶层组，都会在y中对应生成一个结构完全一致的组，完美保留原三维结构。

实际输出验证 #

用你给出的输入x：

x = [ [[1,2,3,4,5],[6,7,8,9,10]], [[11,12,13,14,15],[16,17,18,19,20]], [[21,22,23,24,25],[26,27,28,29,30]], [[21,22,23,24,25]] ]

运行上述代码后，会得到：

• y的结果：

  [ [[1,2,3],[6,7,8]], [[11,12,13],[16,17,18]], [[21,22,23],[26,27,28]], [[21,22,23]] ]
  

• z的结果：

  [ [[4,5],[9,10]], [[14,15],[19,20]], [[24,25],[29,30]], [[24,25]] ]
  

完全符合你的预期需求。

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