迭代 4D numpy 数组的轴数

Question

这可能非常简单，我只是忽略了一些东西，但你可以：

我有一个形状为 (x,y,z,t)

的 4D 数组

如何使用 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