python - Numpy：给定 argmax 的 np.where 函数

Question

我正在尝试在两个 2D numpy 数组上执行此功能：第 1 步：查找np.argmax(b, axis=1)索引。第 2 步：查找b[indices] > a[indices] 第 3 步：在 2D 布尔数组中返回值。

我试过这个：

np.where((b>a)&np.argmax(b,axis=1).reshape((3,-1)), True, False)

但没有骰子。有任何想法吗？

提前致谢。

Answer 1

根据您的评论，我最好的理解是：

output = (np.max(b,axis=1)[...,None] == b) & (b > a)

我们利用 Numpy 广播来执行“是其行中的最大值b”部分：

np.max(b,axis=1)[...,None] == b

或者更清楚：

np.max(b,axis=1)[...,np.newaxis] == b