python - Numpy 迭代 3d 向量数组

Question

我试图迭代一个 Numpy 数组，其中包含 3d numpy 数组(3d 向量)。像这样的事情:

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，而且它也往往会生成更短的代码。