对于 numpy ndarray，Python 循环太慢

Question

我有一个图像作为矩阵 (ndarray) 的形状 (720, 1280, 3) 与 RGB 像素（变量：original）和另一个形状为 (720, 1280) 的 ndarray布尔值（变量：im）。

对于im 中的每个True 值，我希望original 的对应像素颜色为[0, 0, 255]。

我试过了（两个数组都变平了）

for i in range(im.size):
  if(im[i] == True):
    original[i] = [0, 0, 255]

但它太慢了，无法将其作为视频输出。

有什么想法可以加快速度吗？

Answer 1

这和做的一样（没有展平你的数组）：

original[im == True] = [0, 0, 255]

这是你的 for 循环的矢量化表示。