2010-12-27 1 views
0

Например: test_array = numpy.array([[10, 1], [2, 12], [3, 5]]), и я хотел бы что-то сделать по линиям test_array.where(min(test_array[0] * test_array[1])) и вернуть ему ту же структуру, что и numpy.where, указывающий на [10, 1].Как я могу запросить массив numpy, чтобы найти минимальное умножение первых двух столбцов?

ответ

2

, если я не то недоразумение, вы просто хотите numpy.argmin ...

Э.Г.

import numpy as np 
x = np.array([[10, 1], [2, 12], [3, 5]]) 
multi_cols = x[:,0] * x[:,1] 
print x[multi_cols.argmin()] 
Смежные вопросы