如何查询numpy数组以找到前两列的最小乘法？

Question

例如：test_array = numpy.array([[10, 1], [2, 12], [3, 5]])我希望按照test_array.where( min(test_array[0] * test_array[1]) )的方式执行某些操作，并让它返回numpy.where返回指向[10, 1]的相同结构。

Answer 1

除非我误解了某些内容，否则你只需要numpy.argmin ...

E.g。

import numpy as np
x = np.array([[10, 1], [2, 12], [3, 5]])
multi_cols = x[:,0] * x[:,1]
print x[multi_cols.argmin()]