numpy.newaxis的用法

对比

不使用newaxis

import numpy as np
a = np.array([1, 2, 3, 4, 5])
a[0]

结果

1

对单个元素使用newaxis

import numpy as np
a = np.array([1, 2, 3, 4, 5])
a[0, np.newaxis]

结果

array([1])

对所有元素使用newaxis

import numpy as np
a = np.array([1, 2, 3, 4, 5])
a[:, np.newaxis]

结果

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

结论

形象一点讲，在每个矩阵元素再套上一维。不知理解是否正确，敬请指正！