import numpy as np # create a numpy array arr = np.array([0, 1, 0, 3, 5]) # find the indices of non-zero elements indices = np.where(arr) print(indices)