Python Numpy矩阵运算-矩阵[a == b]？

Question

我一直在尝试创建一个分水岭算法，并且所有示例似乎都在Python中，因此我碰到了一堵墙。 我一直试图在numpy文档中找到这行的含义：

matrixVariable[A==255] = 0

但没有运气。 谁能解释该操作的作用？

对于上下文，该行在label [lbl == -1] = 0 ： label [lbl == -1] = 0

Answer 1

表达式A == 255创建一个布尔数组，该数组为True ，其中A中x == 255，否则为False 。

表达式matrixVariable[A==255] = 0将与A == 255中的True值相对应的每个索引设置为0。

例如：

import numpy as np
A = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
B = np.zeros([3, 3])
print('before:')
print(B)
B[A>5] = 5
print('after:')
print(B)

出：

[[ 0.  0.  0.]
 [ 0.  0.  0.]
 [ 0.  0.  0.]]
after:
[[ 0.  0.  0.]
 [ 0.  0.  5.]
 [ 5.  5.  5.]]

Answer 2

我假设matrixVariable和A是numpy数组。 如果假设正确，则“ matrixVariable [A == 255] = 0”表达式首先获取数组A的索引，其中A的值等于255，然后获取这些索引的matrixVariable值并将其设置为“ 0”

例：

import numpy as np

matrixVariable = np.array([(1, 3),
                           (2, 2),
                           (3,1)])

A = np.array([255, 1,255])

所以A [0]和A [2]等于255

matrixVariable[A==255]=0 #then sets matrixVariable[0] and matrixVariable[2] to zero

print(matrixVariable) # this would print 

[[0 0] 
 [2 2] 
 [0 0]]