np.newaxis用法详解

import numpy as np
arr = np.linspace(1, 10, 10)
# print("arr:", arr) 变成了一维数组：arr: [ 1.  2.  3.  4.  5.  6.  7.  8.  9. 10.]

# print(arr[np.newaxis, :]) # [[ 1.  2.  3.  4.  5.  6.  7.  8.  9. 10.]]
# print(arr[:, np.newaxis])
"""
[[ 1.]
 [ 2.]
 [ 3.]
 [ 4.]
 [ 5.]
 [ 6.]
 [ 7.]
 [ 8.]
 [ 9.]
 [10.]]
"""
print(arr.shape) # (10,)
print(arr[np.newaxis, :].shape) # (1, 10)
print(arr[:, np.newaxis].shape)# (10, 1)