最近在学习机器学习的时候,学习到一个为一维数组添加维度的函数np.newaxis
import numpy as np
x = np.arange(10)
x
这是一个普通的一维数组,输出如下:
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
下面我们为其添加维度,有两种方式:
一、使每个元素作为一维
X = x[:, np.newaxis]
X
输出:
array([[0],
[1],
[2],
[3],
[4],
[5],
[6],
[7],
[8],
[9]])
二、将整个数组作为一维
x = x[np.newaxis, :]
x
输出:
array([[[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]]])
后续有更深理解了继续更