创建一个索引数组，其中包含另一个数组中某个值的位置

I am trying to create a Python list that contains indices of the elements equal to 1 in another integer list (or Numpy array). What I am trying is something like this (for either 1- or 2-dimensional case):I am trying to create a Python list that contai