python - numpy数组的逻辑索引

Question

在 Maltab 中，逻辑索引可以按如下方式完成。给定两个二值图像掩码A和B：

C = A(B)

给定两个 numpy 数组A和Bbool 类型，有什么简单的方法可以在 Python 中执行相同的操作吗？

标签： pythonnumpy

Answer 1

假设我们有一个数组并且想要找到大于 50 的值。

代码：

import numpy as np

array = np.array([80, 35, 50, 70])

greater_then_50 = array[array > 50]

print(greater_then_50)

# output:
array([80, 70])

这是 numpy 中的一个示例布尔索引。

每次对 numpy 数组进行布尔运算时，它都会返回一个具有 True 和 False 的布尔数组，当我们将此布尔数组作为掩码传递给另一个数组时，它只会返回布尔数组中具有 True 的位置。