1. 程式人生 > >numpy中的np.newaxis

numpy中的np.newaxis

import numpy as np

x_data=np.linspace(-0.5,0.5,10)
print(x_data)

輸出結果為
[-0.5        -0.38888889 -0.27777778 -0.16666667 -0.05555556  0.05555556
  0.16666667  0.27777778  0.38888889  0.5       ]
十個浮點數,一維度
1行


x_data=np.linspace(-0.5,0.5,10)[:,np.newaxis]
print(x_data)

輸出結果為
[[-0.5       ]
 [-0.38888889]
 [-0.27777778]
 [-0.16666667]
 [-0.05555556]
 [ 0.05555556]
 [ 0.16666667]
 [ 0.27777778]
 [ 0.38888889]
 [ 0.5       ]]
10行1列,二維度

這樣看來,[:,np.newaxis]說白了就是新增一個維度,我們再來試一下

import numpy as np

x=np.ones(3)
print(x)

x=np.ones(3)[:,np.newaxis]
print(x)

x=np.ones(3)[np.newaxis,:]
print(x)

輸出結果為

[1. 1. 1.]
[[1.]
 [1.]
 [1.]]
[[1. 1. 1.]]

意思是np.newaxis加在後邊就是加上列,加在前邊就是再套一行