Python np.newaxis


np.newaxis的功能是插入新維度,看下面的例子:

a=np.array([1,2,3,4,5])
print a.shape

print a

輸出結果

(5,)
[1 2 3 4 5]

可以看出a是一個一維數組,

x_data=np.linspace(-1,1,300)[:,np.newaxis]
a=np.array([1,2,3,4,5])
b=a[np.newaxis,:]
print a.shape,b.shape
print a

print b

輸出結果:

(5,) (1, 5)
[1 2 3 4 5]
[[1 2 3 4 5]]

x_data=np.linspace(-1,1,300)[:,np.newaxis]
a=np.array([1,2,3,4,5])
b=a[:,np.newaxis]
print a.shape,b.shape
print a
print b

輸出結果

(5,) (5, 1)
[1 2 3 4 5]
[[1]
[2]
[3]
[4]
[5]]

可以看出np.newaxis分別是在行或列上增加維度,原來是(6,)的數組,在行上增加維度變成(1,6)的二維數組,在列上增加維度變為(6,1)的二維數組


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM