numpy函数之np.newaxis

np.newaxis的功能是插入新维度。

import numpy as np

a = np.array([1, 2, 3, 4, 5, 6])
print(a.shape)
print(a)

b = a[np.newaxis]
b=a[np.newaxis,:]
print(b.shape)
print(b)

c=a[:,np.newaxis]
print(c.shape)
print(c)

结果：

(6,)
[1 2 3 4 5 6]

(1, 6)
[[1 2 3 4 5 6]]

(6, 1)
[[1]
[2]
[3]
[4]
[5]
[6]]

可以看出np.newaxis分别是在行或列上增加维度，原来是（6，）的数组，在行上增加维度变成（1,6）的二维数组，在列上增加维度变为(6,1)的二维数组

参考文献：

numpy.newaxis

python np.newaxis