使用numpy,是否可以根据其他矩阵定义矩阵?

时间:2017-02-14 15:47:17

标签: python numpy

例如,我想做类似的事情

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

2 个答案:

答案 0 :(得分:5)

一种方法是使用vstackhstack

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

答案 1 :(得分:3)

你几乎可以使用堆叠行的np.r_和堆叠列的np.c_

D = np.r_[np.c_[A, B], np.c_[B.T, C]]