如何将单列numpy数组与四列numpy数组连接?

时间:2015-03-02 17:28:49

标签: arrays numpy concatenation

我有一个numpy数组,它是一个列向量。我有另一个numpy数组,它是与单列向量长度相同的4列向量。如何组合数组为[x,5]?

1 个答案:

答案 0 :(得分:2)

使用numpy.hstack,您可以水平堆叠矢量(当由于某些原因已经以列形状给出矢量时)

import numpy

a = numpy.arange(12).reshape(3,4)
#[[ 0  1  2  3]
# [ 4  5  6  7]
# [ 8  9 10 11]]

b = numpy.arange(3).reshape(3,1)
#[[0]
# [1]
# [2]]

numpy.hstack((a,b))
#[[ 0  1  2  3  0]
# [ 4  5  6  7  1]
# [ 8  9 10 11  2]]

编辑: 否则numpy.stack_column更适合将1d阵列添加到2d阵列,因为Mark不需要重塑,谢谢!