numpy 如何在Python中高效地得到所需形式的矩阵？

b4wnujal 于 2022-12-23 发布在 Python

关注(0)|答案(3)|浏览(107)

我有四个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的单位矩阵。

numpy

来源：https://stackoverflow.com/questions/9655773/how-to-efficiently-get-matrix-of-the-desired-form-in-python

3条答案

按热度按时间

mpbci0fu1#

下面的例子怎么样：

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]])

或者，你可以创建一个适当大小的空数组，然后用适当的切片填充它，这样可以避免创建中间数组。

赞(0）回复(0）举报 2022-12-23

rggaifut2#

您需要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]])

赞(0）回复(0）举报 2022-12-23

du7egjpx3#

对于numpyarray，this page建议语法可以是以下形式

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

赞(0）回复(0）举报 2022-12-23

我来回答

numpy 如何在Python中高效地得到所需形式的矩阵？

3条答案

相关问题

热门标签

最新问答