python - 在 Numpy 数组中查找行最大化某些函数

Question

在最大化某些函数的 Numpy 二维数组中查找行索引的最优雅和最有效的方法是什么？特别是，我需要找到具有给定向量的最大点积的行。说，我有

a = np.array([[1, 2, 3], [3, 2, 1]])
b = np.array([6, 5, 4])

那么结果应该是 1 因为 np.dot(a[1],b) 大于 np.dot(a[0], b).

Answer 1

简单地:

>>> np.argmax(a.dot(b))
1

np.argmax docs