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