如何循环遍历多维矩阵？

Question

所以我想循环遍历一个形状为 (82680, 1, 1024, 4) 的多维矩阵，以便能够标记其中哪些值为 0。我如何才能做到这一点？

Answer 1

这可能不是最有效的方法，但一个简单的方法是

zero_locations = []

for i in range(82680):
    for j in range(1):
        for k in range(1024):
            for l in range(4):
                if matrix[i][j][k][l] == 0:
                    zero_locations.append((i, j, k, l))

这样做是遍历矩阵中的每一项，在最里面的 for 循环中，我们将单元格的值与 0 进行比较。如果单元格等于 0，我们将包含四个“坐标”的元组附加到列表中。