Return index of every non-zero element in array

Question

I know there is a way to return the index of the maximum element in an array in python: numpy.argmax(). Is there a way to return index of every non-zero element?

For example

array([[ 0.,  1.,  1., ...,  1.,  0.,  0.],
       [ 0.,  1.,  1., ...,  1.,  0.,  1.],
       [ 0.,  1.,  1., ...,  1.,  0.,  0.],
       ..., 
       [ 0.,  1.,  1., ...,  1.,  0.,  0.],
       [ 0.,  1.,  1., ...,  1.,  0.,  0.],
       [ 0.,  1.,  1., ...,  1.,  0.,  0.]], dtype=float32)

to

[[1, 2, ...,6],
[1,2,...6,8],
...
...
]

antikantian · Accepted Answer · 2016-02-27 03:29:05Z

2

Do you want something like this:

x = np.asarray([0, 1, 2, 3, 0, 1])
In [129]: np.nonzero(x)
Out[129]: (array([1, 2, 3, 5]),)

See: [1], [2], [3]

answered Feb 27, 2016 at 3:29

antikantian

6401 gold badge5 silver badges11 bronze badges

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

1 Comment

Jill Clover Over a year ago

Yes. I cannot believe why I did not get right answer from google.

Collectives™ on Stack Overflow

Return index of every non-zero element in array

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related