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中，並且也傾向於生成較短的代碼。