在Python中的整数转换中的二进制数组

Question

我有一个矩阵，格式为：

 a=[[1 0 0 0 0],
  [0 0 1 0 0],
  [0 1 0 0 0],
  [0 0 0 1 0],
  [0 0 0 0 1]]

，我想将向量 a 转换为以下形式的矩阵：

 b=[[0],[2],[1],[3],[4]]

因此，基本上，我想将二进制矩阵转换为整数矩阵，其条目对应于 a 矩阵中的“ 1”的索引/位置。

您是否知道它在python中如何工作？