如何从数组中选择一个数字并告诉您Python中的列和行？

Question

If I have, for example, an array of 5 * 5 that has as numbers the list from 1 to 25 such that:

X = list (range (1,26))

Dimension = 5
A = np.zeros ((dimension, dimension))
N = 0
For i in range (dimension):
     For j in range (dimension):
         A [i] [j] = x [n]
         N = n + 1

And I want python tell me the column and row where this number 16, how is it done?

Answer 1

To find 16 in :

a = [[  1.   2.   3.   4.   5.]
     [  6.   7.   8.   9.  10.]
     [ 11.  12.  13.  14.  15.]
     [ 16.  17.  18.  19.  20.]
     [ 21.  22.  23.  24.  25.]]

Do:

np.where(a == 16)

Output:

(array([3]), array([0]))