numpy 3d array — flatten --> 1d array --> select one element in 1d --> how to know the index of the element in 3d?

Question

numpy 3d array -- flatten --> 1d array --> select one element in 1d array --> how to know the index of the element in 3d? (?, ?, ?)

<