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!
Answers:
Use unravel_index:
numpy.unravel_index(A.argmin(), A.shape)
[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
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!
Use unravel_index:
numpy.unravel_index(A.argmin(), A.shape)
[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