迭代 4D numpy 数组的轴数

Question

这可能很简单，我只是忽略了一些东西，但是你去吧：

我有一个形状为 4D 的数组：(x,y,z,t)

如何使用 for 循环遍历该数组的轴数？不是沿单个轴，而是每个轴一次？

类似这样的：

for a in range(data.shape):
            data_20 = np.percentile(data_20, 80, axis=a, keepdims=True)

谢谢！

Answer 1

要遍历numpy.array 的维度范围，请使用.ndim

代码：

# convert data to numpy.array if data is a list

for i in range(np.array(data).ndim):
    print(i)

输出：

0
1
2
3