关于np.newaxis的理解

np.newaxis的作用就是在这一位置增加一个维度,这一位置指的是np.newaxis所在的位置

x1 = np.array([1, 2, 3, 4, 5])
# the shape of x1 is (5,)

这里x1的shape为(5,) ,这里我们用newaxis,改变维度

x1_new = x1[:, np.newaxis]
# now, the shape of x1_new is (5, 1)
# array([[1],
#        [2],
#        [3],
#        [4],
#        [5]])
x1_new = x1[np.newaxis,:]
# now, the shape of x1_new is (1, 5)
# array([[1, 2, 3, 4, 5]])

 接着往外引申一下:

# promoting 1D array to a 3D array
x1_3D = x1[np.newaxis,:, np.newaxis]
x1_3D.shape
#(1, 5, 1)

同时我改变维度还可以用另一个方法:

shape 

x1.shape = (5,1)
# now, the shape of x1 is (5, 1)
# array([[1],
#        [2],
#        [3],
#        [4],
#        [5]])
发布了67 篇原创文章 · 获赞 48 · 访问量 4万+

猜你喜欢

转载自blog.csdn.net/qq_41282102/article/details/104201769