迭代多维 numpy 数组答案

【问题标题】：Iterating over multidimensional numpy arrays迭代多维 numpy 数组
【发布时间】：2019-03-30 20:54:50
【问题描述】：

我有一个尺寸为 (a,b,c) 的 numpy 数组。我想把它切成c'th维度。例如：一个形状为 (2,3,4) 的 numpy 数组，我想遍历 4 个维度为 (2,3) 的数组。

到目前为止我一直在做

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

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

【问题讨论】：

【解决方案1】：

冒号之间不是也需要逗号吗？像这样： arr = A[:, :, i]

【讨论】：

【解决方案2】：

您缺少一个逗号。您的代码应如下所示：

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

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

【讨论】：