python中numpy矩阵如何增维--newaxis的用法

python中numpy矩阵如何增维–newaxis的用法

把测试集放入分类树，回归树时，必须是二维的，所以遇到一维数据时，应该对数据进行增维，那么该如何增维呢？

创建一个一维矩阵

import numpy as np
a = np.array([1,2,3,4])
b = a.shape
print(a)
print(b)

[1 2 3 4]
(4,)

增维成 —>四行一列

b = a[:,np.newaxis]
print(b.shape)
print(b)

(4, 1)
[[1]
 [2]
 [3]
 [4]]

增维成—>一行四列

c = a[np.newaxis,:]
print(c.shape)
print(c)

(1, 4)
[[1 2 3 4]]