Python 如何在一个NumPy数组中搜索位置，以便从第二个NumPy数组中获取这些位置的值？

我有两个

光栅

文件，我已将它们转换为

NumPy

数组（

arcpy.RasterToNumpyArray

），以便使用Python处理

光栅

单元格中的值

其中一个光栅有两个值

True

和

False

。另一个光栅在

0

到

1000

之间具有不同的值。两个

光栅

具有完全相同的范围，因此除了值之外，两个

NumPy

阵列的构建方式相同（列和行）

我的目标是识别

NumPy

数组

A

中具有值

True

的所有位置。这些位置用于从

NumPy

数组

B

获取这些位置的值

---

你知道我如何实现这一点吗？

如果我正确理解你的描述，你应该能够执行

B[A]

你可以使用带有

True

和

False

值的数组来简单地索引到另一个数组中。以下是一个示例：

import numpy as np
a = np.array([[1,2,3],[4,5,6],[7,8,9]])
b = np.array([[True,False,False],[False,True,False],[False,False,True]])
a[b] ## gives  array([1, 5, 9])