遍历多维numpy数组

时间:2019-03-30 12:34:11

标签: python numpy

我有一个尺寸为(a,b,c)的numpy数组。我想将其切成第c个维度。例如:一个形状为(2,3,4)的numpy数组,我想遍历4个尺寸为(2,3)的数组。

到目前为止我一直在做

nix search

但这不能算是正确的事情。我该如何计算?

2 个答案:

答案 0 :(得分:1)

在冒号之间也要用逗号吗?像这样: arr = A [:,:,i]

答案 1 :(得分:0)

您缺少逗号。您的代码应如下所示:

for i in range(c):
    arr = A[:,:,i]

顺便说一句,上面的代码正在计算正确的东西,但是您编写的语句无法解决问题:)