numpy中的np.newaxis
阿新 • • 發佈:2018-11-14
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加在後邊就是加上列,加在前邊就是再套一行