python - 使用 numpy，是否可以根据其他矩阵定义矩阵？

Question

例如，我想做这样的事情

import numpy as np
A = np.array([[1,2,3],[4,5,6],[7,8,9]])
B = np.array([[1,1],[2,2],[3,3]])
C = np.zeros((2,2))
D = np.array([[A, B], [B.T, C]])

生成一个包含元素的 5x5 ndarray

1, 2, 3, 1, 1
4, 5, 6, 2, 2
7, 8, 9, 3, 3
1, 2, 3, 0, 0
1, 2, 3, 0, 0

Answer 1

一种方法是使用 vstack 和 hstack:

D = np.vstack((np.hstack((A, B)), np.hstack((B.T, C))))