numpy里的np.newaxis有什么用？

newaxis是可以对numpy array进行升维操作的。

arr = np.arange(5)
# arr的shape是(5, )

# 1d升到2d
arr1 = arr[np.newaxis, :]
# arr1的shape是(1, 5)

# 1d升到3d
arr2 = arr[np.newaxis, np.newaxis, :]
# arr2的shape是(1, 1, 5)

np.newaxis就是保持原数据的情况凭空增加一个维度