numpy 中 newaxis函数的使用

来源:互联网 发布:其皆出于此乎句式 编辑:程序博客网 时间:2024/06/01 08:50

numpy 中 newaxis函数的使用

newaxis表示增加一个新的坐标轴,不好理解,看例子就明白了

第一个程序

import numpy as npa = np.array([1,2,3])print (a.shape,'\n',a)

结果为:
(3,)
[1 2 3]

看下面的:

第二个程序

a = np.array([1,2,3])[:,np.newaxis]print (a.shape,'\n',a)

(3, 1)
[[1]
[2]
[3]]
你会发现,和第一个程序相比,a的shape为(3,)现在为(3,1)变为二维数组了,之前为[1,2,3],现在变为 [[1]
[2]
[3]]

第三个程序

a = np.array([1,2,3])[np.newaxis,:]print (a.shape,'\n',a)

输出结果为:
(1, 3)
[[1 2 3]]
这个和第二个相比,好像和他是反的,相当于转置了,这是因为和[np.newaxis,:]这个地方np.newaxis放的位置有关,第二个程序放在[:,]的后面,相当于在原来的后面增加一个维度,所以变为(3,1),而第三个则放在前面,则为(1,3),记得注意啊,放在前面是先逗号,在冒号,而放在后面是先冒号在逗号,不要弄错了哦,同时记得是中括号扩起来,不是小括号哦
## 总结
np.newaxis的作用就是在原来的数组上增加一个维度。

原创粉丝点击