如何在另一个矩阵中间插入一个矩阵增加大小，同时保持旧矩阵元素之间的行列关系

如何解决如何在另一个矩阵中间插入一个矩阵增加大小，同时保持旧矩阵元素之间的行列关系

假设我们有一个矩阵 A，

A=[
   a 0 0 b
   0 c d 0
   0 e f 0
   g 0 0 h
  ]

我想要一种有效且通用的方法，在沿对角线的某个指定索引处添加一些大小（例如 2x2）单位矩阵，例如在 c 之后，同时保持 A 中显示的元素之间的所有关系。最终矩阵应该是

B=[
   a 0 0 0 0 b
   0 c 0 0 d 0
   0 0 1 0 0 0
   0 0 0 1 0 0
   0 e 0 0 f 0
   g 0 0 0 0 h
  ]

我已经在 scipy.linalg 中尝试了 block_diag，但我找不到让它工作的方法