如何循环遍历多维矩阵？

Question

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

Answer 1

This might not be the most efficient way to go about this, but an easy way to do this would be

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))

What this does is iterate through every item in the matrix, and within the innermost for loop we compare the value of the cell to 0. If the cell is equal to zero, we append a tuple containing the four 'coordinates' to a list.