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

Question

假设我有一个二维的numpy数组

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

我有两个布尔数组，一个用于列，一个用于行。

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

现在给定这两个数组，我只想在2d数组中找到只有列和行均为真的那些索引中的argmin。 我可以使用for循环来做到这一点，但是有一种“ numpy”方式吗？ 谢谢

Answer 1

此功能可以解决您的问题（获取索引）

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