Numpy迭代3d向量数组

Question

我试图遍历其中包含3d numpy数组（3d向量）的Numpy数组。 像这样：

import numpy as np


Matrix = np.zeros(shape=(10, 3))
# => [
    [0,0,0],
    [0,0,0],
    ...
    [0,0,0]
]

我需要遍历它，获取每个3d矢量。 用伪代码：

for vector in Matrix
    print vector #=> [0,0,0]

是否有任何Numpy本机方式？ 最快的方法是什么？

谢谢！

弗兰

Answer 1

您的伪代码仅缺少冒号：

for vector in matrix:
    print vector

也就是说，您通常希望避免在NumPy数组上进行显式迭代。 充分利用广播操作和NumPy内置功能； 它将循环移动到C而不是解释的Python中，并且也倾向于生成较短的代码。