np.newaxis
最近看到np.newaxis
这方法,思考了一下午这个如何理解,于是我试验出来了。
结果就是,若现在有一个代码:
diabetes_X[:, np.newaxis, 2]
它的过程等价于:
diabetes_X[:, 2].reshape(:, 1, 2)
因为我一直在思考它如何变化的,它的方法介绍是增加一个轴,但是当我们指定了一个轴切片时便不好理解了。
比如:
arr = arange(3)
arr = np.array([arr, arr])
arr = arr[:,:,np.newaxis]
此时,根据它的意思,我们将一个(2,3)
形状的数组,变成了一个(2,3,1)
的数组。那么如果arr = arr[:,:,np.1]
这样的代码就不太好理解了,所以才有了转换的思考过程。