对比
不使用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]])
结论
形象一点讲,在每个矩阵元素再套上一维。不知理解是否正确,敬请指正!