Numpy minimum in (row, column) format

Question

How can I know the (row, column) index of the minimum of a numpy array/matrix?

For example, if A = array([[1, 2], [3, 0]]), I want to get (1, 1)

Thanks!

Philipp · Accepted Answer · 2010-07-12 16:13:13Z

157

numpy.unravel_index(A.argmin(), A.shape)

answered Jul 12, 2010 at 16:13

Philipp

50.1k12 gold badges88 silver badges112 bronze badges

Sign up to request clarification or add additional context in comments.

3 Comments

Thanks, but to comment this is surely a incredibly complex syntax for doing a simple stuff.

it really is quite standard syntax, simple even...its just one function extra function call beyond A.argmin(), which returns an index....

@linello, just create a function and reuse it, the complexity of the syntax doesn't matter

Nikesh Bajaj · Accepted Answer · 2019-01-20 19:46:30Z

4

[Corrected typo]

Another simple solution is

ri, ci = A.argmin()//A.shape[1], A.argmin()%A.shape[1]

As numpy.argmin returns the index reading in row-major order

Yes, you are right, it was a typo, which worked for square matrix

answered Mar 27, 2018 at 8:25

Nikesh Bajaj

794 bronze badges