Аппликационная матрица A с матрицей B

Question

Скажем, у меня есть две матрицы A и B. Например,

A = numpy.zeros((5,5)) 

B = np.eye(5) 

Есть ли способ, чтобы добавить A и B?

Answer 1

Это звучит для меня, как вы ищете np.hstack:

>>> import numpy as np 
>>> a = np.zeros((5, 5)) 
>>> b = np.eye(5) 
>>> np.hstack((a, b)) 
array([[ 0., 0., 0., 0., 0., 1., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0., 0., 1., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0., 0., 0., 1., 0., 0.], 
     [ 0., 0., 0., 0., 0., 0., 0., 0., 1., 0.], 
     [ 0., 0., 0., 0., 0., 0., 0., 0., 0., 1.]]) 

np.vstack будет работать, если вы хотите, чтобы сложить их вниз:

>>> np.vstack((a, b)) 
array([[ 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0.], 
     [ 0., 0., 0., 0., 0.], 
     [ 1., 0., 0., 0., 0.], 
     [ 0., 1., 0., 0., 0.], 
     [ 0., 0., 1., 0., 0.], 
     [ 0., 0., 0., 1., 0.], 
     [ 0., 0., 0., 0., 1.]])