在二维数组中查找特定索引的 argmin

Question

假设我有一个二维 numpy 阵列

[[1,2,3]
[4,5,6]
[7,8,9]]

我有两个 boolean arrays，一个用于列，一个用于行。

a = [False,True,False]
b = [True, False, True]

现在给定这两个 arrays 我想在二维数组中找到仅列和行都为真的那些索引的 argmin。 我可以使用 for 循环来做到这一点，但有没有“numpy”的方式？ 谢谢

Answer 1

这个 function 解决了你的问题（获取索引）

numpy.where(#Your Condition#) #for your example condition is a==b