import numpy as np
x = np.array([1,2,3])
y = x[:,np.newaxis]
z= x[np.newaxis,:]
w = x[np.newaxis]
print(x.shape,y.shape,z.shape,w.shape)
#输出 ((3,), (3, 1), (1, 3), (1, 3))
可以看出newaxis用来增加一个维度,新增的维度的长度大小默认为1,放在左边就代表新增一行,列数用原来的替代。
numpy中的newaxis用法
最新推荐文章于 2024-03-14 09:13:15 发布