python实现数字和数组合并

将两个数组合并,可以简单的通过np.stack((a,b),axis=1)或者np.concatenate((a,b),axis=1)实现
但是想实现数字c与数组d的合并,比如:

a = 1
b = [2,3,4,5,6]  #shape:(3,)
c = [1,2,3,4,5,6]

将a和b合并为c,直接使用np.stack((a,b),axis=-1)会报错:

ValueError: all input arrays must have the same shape

考虑到a此时是0维,而b的数据集中在第0维度,手动给a增加维度可以实现合并

np.stack(([a],b),axis=0)

你可能感兴趣的:(Python,python,numpy,开发语言)