在numpy中附加2x2协方差矩阵

Question

我有一个numpy数组，例如：

你知道吗gmm.sigma公司=

[[[ 4.64 -1.93]
  [-1.93  6.5 ]]
 [[ 3.65  2.89]
  [ 2.89 -1.26]]]

我想添加另一个2x2矩阵，例如：

你知道吗高斯西格玛=你知道吗

[[ -1.24  2.34]
 [  2.34  4.76]]

要获得：

你知道吗gmm.sigma公司=

[[[ 4.64 -1.93]
  [-1.93  6.5 ]]
 [[ 3.65  2.89]
  [ 2.89 -1.26]]
 [[-1.24  2.34]
  [ 2.34  4.76]]]

我试过：gmm.sigma = np.append(gmm.sigma, gauss.sigma, axis = 0)， 但是得到这个错误：

Traceback (most recent call last):
  File "test1.py", line 40, in <module>
    gmm.sigma = np.append(gmm.sigma, gauss.sigma, axis = 0)
  File "/home/rowan/anaconda2/lib/python2.7/site-packages/numpy/lib/function_base.py", line 4528, in append
    return concatenate((arr, values), axis=axis)
ValueError: all the input arrays must have same number of dimensions

感谢您的帮助

Answer 1

您可以使用^{}将数组按深度顺序（沿第三个轴）堆叠，然后进行转置。要获得所需的输出，必须对gmm.T和gauss进行堆栈

gmm = np.array([[[4.64, -1.93],
                [-1.93, 6.5 ]],
                [[3.65, 2.89],
                 [2.89, -1.26]]])

gauss = np.array([[ -1.24, 2.34],
                  [2.34, 4.76]])


result = np.dstack((gmm.T, gauss)).T
print (result)
print (result.shape)
# (3, 2, 2)

---

输出

array([[[ 4.64, -1.93],
    [-1.93,  6.5 ]],

   [[ 3.65,  2.89],
    [ 2.89, -1.26]],

   [[-1.24,  2.34],
    [ 2.34,  4.76]]])

或者您也可以使用concatenate，将第二个数组正确地重新格式化为

gmm = np.array([[[4.64, -1.93],
                [-1.93, 6.5 ]],
                [[3.65, 2.89],
                 [2.89, -1.26]]])

gauss = np.array([[ -1.24, 2.34],
                  [2.34, 4.76]]).reshape(1,2,2)

result = np.concatenate((gmm, gauss), axis=0)