Question

我有四个numpy数组，如：

X1 = array([[1, 2], [2, 0]])

X2 = array([[3, 1], [2, 2]])

I1 = array([[1], [1]])

I2 = array([[1], [1]])

我正在做：

Y = array([I1, X1],
          [I2, X2]])

获得：

Y = array([[ 1,  1,  2],
           [ 1,  2,  0],
           [-1, -3, -1],
           [-1, -2, -2]])

就像这个例子一样，我有大型矩阵，其中X1和X2是n x d矩阵。

在Python中有一种有效的方法，我可以得到矩阵Y吗？

虽然我知道迭代方式，但我正在寻找一种有效的方法来完成上述工作。

此处，Y是n x (d+1)矩阵，I1和I2是维度n x 1的标识矩阵。

Answer 1

以下内容如何：

In [1]: import numpy as np

In [2]: X1 = np.array([[1,2],[2,0]])

In [3]: X2 = np.array([[3,1],[2,2]])

In [4]: I1 = np.array([[1],[1]])

In [5]: I2 = np.array([[4],[4]])

In [7]: Y = np.vstack((np.hstack((I1,X1)),np.hstack((I2,X2))))

In [8]: Y
Out[8]: 
array([[1, 1, 2],
       [1, 2, 0],
       [4, 3, 1],
       [4, 2, 2]])

或者，您可以创建一个适当大小的空数组，并使用适当的切片填充它。这样可以避免制作中间数组。

Answer 2

您需要numpy.bmat

In [4]: A = np.mat('1 ; 1 ')
In [5]: B = np.mat('2 2; 2 2')
In [6]: C = np.mat('3 ; 5')
In [7]: D = np.mat('7 8; 9 0')
In [8]: np.bmat([[A,B],[C,D]])
Out[8]: 
matrix([[1, 2, 2],
        [1, 2, 2],
        [3, 7, 8],
        [5, 9, 0]])

Answer 3

对于numpy array，this page建议语法可以是

形式

vstack([hstack([a,b]),
        hstack([c,d])])

如何在Python中有效地获取所需形式的矩阵？

3 个答案: