迭代多维numpy数组

Question

我有一个numpy维数（a，b，c）。我想把它切成第c维。例如：一个numpy数组形状（2,3,4），我想迭代4个维数阵列（2,3）。

到目前为止，我一直在做

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

但这并不能算出正确的事情。我该如何计算？

Answer 1

你错过了一个逗号。您的代码应如下所示：

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

顺便说一下上面的代码是计算正确的东西，但你正在写一个不能解决问题的声明:)

Answer 2

冒号之间是否也有逗号？像这样：arr = A [：，：，i]