python - 基于另一个数组中的信息对 NumPy 数组进行操作

Question

我对图像进行了均值漂移分割并得到了标签数组，其中每个点值对应于它所属的片段。

labels = [[0,0,0,0,1],
          [2,2,1,1,1],
          [0,0,2,2,1]]

另一方面，我有相应的grayScale图像，想对每个区域独立进行操作。

img = [[100,110,105,100,84],
       [ 40, 42, 81, 78,83],
       [105,103, 45, 52,88]]

比方说，我想要每个区域的灰度值之和，如果它小于 200，我想将这些点设置为 0(在这种情况下，区域 2 中的所有点)，我该怎么做与 NumPy ？我确信有比我已经开始的实现更好的方法，它包括很多很多 for 循环和临时变量...

Answer 1

查看 numpy.bincount 和 numpy.where，这应该可以帮助您入门。例如:

import numpy as np
labels = np.array([[0,0,0,0,1],
                   [2,2,1,1,1],
                   [0,0,2,2,1]])
img = np.array([[100,110,105,100,84],
                [ 40, 42, 81, 78,83],
                [105,103, 45, 52,88]])

# Sum the regions by label:
sums = np.bincount(labels.ravel(), img.ravel())

# Create new image by applying threhold
final = np.where(sums[labels] < 200, -1, img)
print final
# [[100 110 105 100  84]
#  [ -1  -1  81  78  83]
#  [105 103  -1  -1  88]]