numpy.AxisError: axis 1 is out of bounds for array of dimension 1

原文地址

错误发生在执行np.concatenate(,axis=1)时

当我想把两个一维数据堆起来的时候，即

# 把
a = [1,2,3]
b = [4,5,6]
# 变成
[[1,2,3],
[4,5,6]]

使用np.concatenate((a,b),axis=1)

这是因为a和b都是一维数据，只有一个维度，也就是axis=0，不存在axis=1

我找到两种解决方案

• np.vstack((A,B))

可以对A，B在竖向进行堆叠

print(np.vstack((a,b)))	# 注意传的参数为‘(a,b)’
# [[1 2 3] 
#  [4 5 6]]

美中不足的是这个方法只能传两个向量进行堆叠

• np.newaxis+np.concatenate()

newaxis，顾名思义就是一个新的轴，用法如下

a = a[np.newaxis,:]	# 其中‘:’代表所有维度（这里是3），a的形状就变成了（1，3），这是个二维的了
# [[1 2 3]]
b = b[np.newaxis,:]
# [[4 5 6]]

这时候我就可以把2个（1，3）的向量堆成（1*2，3）=（2，3）的矩阵了，注意这时候就应该用axis=0了，也就是第1个维度

print(np.concatenate((a,b),axis=0))
# [[1 2 3] 
#  [4 5 6]]

相关

Numpy：矩阵合并