迭代numpy数组的最后一个维度

Question

考虑以下 sn-p：

a = np.ones((2,2,2,2,2))
for example in a:
    for row in example:
        for col in row:
            for box in col:
                print (box.shape)

有这么多嵌套的fors 会导致一些非常丑陋的代码。

如何仅通过一次显式迭代获得相同的效果？

Answer 1

重塑你的数组：

for box in a.reshape(16,2):
    print(box.shape)

---

一个通用的解决方案：

for box in a.reshape(np.prod(a.shape[:-1]), a.shape[-1]):
     print(box.shape)

它将始终遍历a的最后一个维度。