numpy 中 newaxis函数的使用

numpy 中 newaxis函数的使用

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

第一个程序

import numpy as np
a = 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的作用就是在原来的数组上增加一个维度。