import numpy as np # create a numpy array arr = np.array([0, 1, 0, 4, 6, 7]) # find the indices of non-zero elements indices = np.argwhere(arr) print(indices)