python数组添加列,如何向NumPy数组添加额外的列

np.r_[ ... ]和np.c_[ ... ]

是vstack和hstack的有用替代品，

用方括号[]代替圆括号()。

举几个例子：: import numpy as np

: N = 3

: A = np.eye(N)

: np.c_[ A, np.ones(N) ] # add a column

array([[ 1., 0., 0., 1.],

[ 0., 1., 0., 1.],

[ 0., 0., 1., 1.]])

: np.c_[ np.ones(N), A, np.ones(N) ] # or two

array([[ 1., 1., 0., 0., 1.],

[ 1., 0., 1., 0., 1.],

[ 1., 0., 0., 1., 1.]])

: np.r_[ A, [A[1]] ] # add a row

array([[ 1., 0., 0.],

[ 0., 1., 0.],

[ 0., 0., 1.],

[ 0., 1., 0.]])

: # not np.r_[ A, A[1] ]

: np.r_[ A[0], 1, 2, 3, A[1] ] # mix vecs and scalars

array([ 1., 0., 0., 1., 2., 3., 0., 1., 0.])

: np.r_[ A[0], [1, 2, 3], A[1] ] # lists

array([ 1., 0., 0., 1., 2., 3., 0., 1., 0.])

: np.r_[ A[0], (1, 2, 3), A[1] ] # tuples

array([ 1., 0., 0., 1., 2., 3., 0., 1., 0.])

: np.r_[ A[0], 1:4, A[1] ] # same, 1:4 == arange(1,4) == 1,2,3

array([ 1., 0., 0., 1., 2., 3., 0., 1., 0.])

(方括号[]而不是圆()的原因)

Python是以1:4的比例展开的--

超载的奇迹。)