Get the index that meets the conditions from the 2d array

get_indices_in_conditions_from_2d_array.py

import numpy as np

# sample array
array = np.array([[1, 1, 1, 0, 0, 0],
                 [1, 1, 1, 0, 0, 0]])
# get row index and col index
rows, cols = np.where(array == 1)
# Convert row index and column index to 2d array
np.transpose((rows, cols))
# array([[0, 0],
#        [0, 1],
#        [0, 2],
#        [1, 0],
#        [1, 1],
#        [1, 2]])