错误发生在执行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]]
相关