Coordinates of item on NumPy array

后端 未结 2 1369
暖寄归人
暖寄归人 2021-02-05 09:18

I have a NumPy array:

[[  0.   1.   2.   3.   4.]
 [  7.   8.   9.  10.   4.]
 [ 14.  15.  16.  17.   4.]
 [  1.  20.  21.  22.  23.]
 [ 27.  28.   1.  20.  29.]         


        
2条回答
  •  再見小時候
    2021-02-05 10:03

    a = numpy.array([[  0.,  1.,  2.,  3.,  4.],
                     [  7.,  8.,  9., 10.,  4.],
                     [ 14., 15., 16., 17.,  4.],
                     [  1., 20., 21., 22., 23.],
                     [ 27., 28.,  1., 20., 29.]])
    print numpy.argwhere(a == 4.)
    

    prints

    [[0 4]
     [1 4]
     [2 4]]
    

    The usual caveats for floating point comparisons apply.

提交回复
热议问题