添加维度numpy数组

时间:2019-03-25 14:32:14

标签: arrays numpy

我有两个numpy数组,例如:

a = np.array([1, 2, 3])

b = np.array([101, 102, 103 ])

我想创建一个形状为(len(a),2)的新数组,例如

array([[1, 101], [2, 102], [3, 103]])

如何用numpy做到这一点?

2 个答案:

答案 0 :(得分:2)

这就是所谓的column_stack

np.column_stack((a,b))
Out[309]: 
array([[  1, 101],
       [  2, 102],
       [  3, 103]])

答案 1 :(得分:1)

或者:

np.c_[a,b]

也可以做。