Numpy 遍历三维向量数组

Question

我正在尝试遍历一个包含三维Numpy数组（也就是三维向量）的Numpy数组。

import numpy as np


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

我需要逐个获取每个三维向量。用伪代码来说就是：

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

有没有Numpy自带的方法可以做到这一点？ 哪种方法最快呢？

谢谢！

Fran

Answer 1

你的伪代码只缺少一个冒号：

for vector in matrix:
    print vector

不过，通常来说，你应该尽量避免直接对NumPy数组进行循环操作。尽量利用广播操作和NumPy自带的函数，这样可以把循环的工作交给C语言来处理，而不是用Python来解释执行，这样不仅能让代码更简洁，还能提高运行速度。