Say I have two matrices A and B. For example,
JavaScript
x
4
1
A = numpy.zeros((5,5))2
3
B = np.eye(5)4
Is there a way to append A and B?
Advertisement
Answer
It sounds to me like you’re looking for np.hstack:
JavaScript
1
10
10
1
>>> import numpy as np2
>>> a = np.zeros((5, 5))3
>>> b = np.eye(5)4
>>> np.hstack((a, b))5
array([[ 0., 0., 0., 0., 0., 1., 0., 0., 0., 0.],6
[ 0., 0., 0., 0., 0., 0., 1., 0., 0., 0.],7
[ 0., 0., 0., 0., 0., 0., 0., 1., 0., 0.],8
[ 0., 0., 0., 0., 0., 0., 0., 0., 1., 0.],9
[ 0., 0., 0., 0., 0., 0., 0., 0., 0., 1.]])10
np.vstack will work if you want to stack them downward:
JavaScript
1
12
12
1
>>> np.vstack((a, b))2
array([[ 0., 0., 0., 0., 0.],3
[ 0., 0., 0., 0., 0.],4
[ 0., 0., 0., 0., 0.],5
[ 0., 0., 0., 0., 0.],6
[ 0., 0., 0., 0., 0.],7
[ 1., 0., 0., 0., 0.],8
[ 0., 1., 0., 0., 0.],9
[ 0., 0., 1., 0., 0.],10
[ 0., 0., 0., 1., 0.],11
[ 0., 0., 0., 0., 1.]])12