1. 程式人生 > >關於np.newaxis的一點理解

關於np.newaxis的一點理解

經常在sklearn上看到np.newaxis,這裡記錄一下我的理解

np.arange(0, 10)

這句話 生成的是一個一維的陣列,如下:
[0 1 2 3 4 5 6 7 8 9]

輸出其shape:(10,)

那麼我如何才能將其轉化為shape=(1,10)呢

可以用兩種方法:

1.使用shape

y=np.arange(1, 11)
y.shape=(10,1)
print(y)

結果如下:
[[ 1]
 [ 2]
 [ 3]
 [ 4]
 [ 5]
 [ 6]
 [ 7]
 [ 8]
 [ 9]
 [10]]

2. 使用np.newaxis
print(np.arange(0, 10)[:, np.newaxis])

結果如下:
[[0]
 [1]
 [2]
 [3]
 [4]
 [5]
 [6]
 [7]
 [8]
 [9]]

如上所示,應該清楚了np.newaxis了吧

上面的程式碼實質就是將原本的(10,)移到行上,然後新增一列