矩阵列表的块对角线

Question

我想用block_diag来制作多个矩阵的块对角线矩阵 。

比方说，我有矩阵A,B和C。我可以简单地写：

block_diag(A, B, C) 

但是，我确实希望在以后有更多的矩阵。

如何合并A，B和C成矩阵X的列表，以便 我可以这样写：

block_diag(X) 

我尝试使用[A, B, C]但不起作用。

Answer 1

使用星号运算符（*），它将iterable的内容展开为位置函数参数。 （我选择使用一个不变的tuple，但名单[A, B, C]会工作得同样好。）

X = (A, B, C) 

print(block_diag(*X))