How to find the indices of a specific element in a numpy array with Python

1 Answer

0 votes
import numpy as np
 
arr = np.array([1, 1, 2, 3, 1, 0, 0, 4, 5, 1, 7, 8, 1, 5, 1, 1, 1])

indices = np.where(arr==1)[0]

print(indices)




'''
run:

[ 0  1  4  9 12 14 15 16]

'''

 



answered Mar 23, 2023 by avibootz
...