python - 将 numpy 数组堆叠到对角线上

Question

给定 N 个 2d numpy 数组，是否有一种巧妙的方法可以让我在对角线上“堆叠”或“连接”它们，用 0 填充任何新的槽？例如。给出:

arr1 = np.array([[1, 2],
                 [3, 4]])

arr2 = np.array([[9, 8, 7],
                 [6, 5, 4],
                 [3, 2, 1]])

我要创建:

arr = np.array([[1, 2, 0, 0, 0],
                [3, 4, 0, 0, 0], 
                [0, 0, 9, 8, 7],
                [0, 0, 6, 5, 4],
                [0, 0, 3, 2, 1]])

Answer 1

There's a function for that.

scipy.linalg.block_diag(arr1, arr2)

它接受任意多个参数:

scipy.linalg.block_diag(*list_of_arrays)