迭代 4D numpy 数组的轴数

问题描述

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

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

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

像这样：

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

谢谢！

解决方法

要迭代 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