为numpy数组增加一个维度的方法

skearn中要求输入的特征数组至少二维,而有时特征数组不符合要求,需要增维

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

方法

0.利用数组长度扩维

b=a.reshape(((len(a),1)))#(3,1)

1.直接扩维

c=a.reshape(-1,1)#(3,1)

-1的意义:根据另一个参数的维度计算出数组的另外一个shape属性值。

[[1]
 [2]
 [3]]
c=a.reshape(1,-1)
[[1 2 3]]

2.生成新的维度

利用np.newaxis

d=a[:,np.newaxis]
[[1]
 [2]
 [3]]

或者

d=a[np.newaxis,:]
[[1 2 3]]
发布了24 篇原创文章 · 获赞 8 · 访问量 2163

猜你喜欢

转载自blog.csdn.net/weixin_44839513/article/details/103869840