作用
np.newaxis的作用是帮助数组创建新轴,或者也叫增加维度。np.newaxis 在使用和功能上等价于 None,其实就是 None 的一个别名。
a= np.array([x for x in range(5)])
print(a)
print(a.shape)
[0 1 2 3 4]
(5,)
b = a[np.newaxis,:]
print(b)
print(b.shape)
[[0 1 2 3 4]]
(1, 5)
c = a[:,np.newaxis]
print(c)
print(c.shape)
[[0]
[1]
[2]
[3]
[4]]
(5, 1)
版权声明:本文为qq_20135597原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。