1. 程式人生 > 程式設計 >python [:3] 實現提取陣列中的數

python [:3] 實現提取陣列中的數

搜尋答案搜尋不到,自己試了一把.

首先生成一維陣列

a =np.array([1,2,3,4,5,6,7,8,9])
>>> print a
[1 2 3 4 5 6 7 8 9]

取陣列前3個值

b =a[:3]
>>> print b
[1 2 3]

取前3個以後的值

b =a[3:]
>>> print b
[4 5 6 7 8 9]

取陣列的後3個值

b =a[-3:]
>>> print b
[7 8 9]

取陣列後3個以前的值

b =a[:-3]
>>> print b
[1 2 3 4 5 6]

所以-號表示方向,從前取還是從後取,與數字配合使用,:表示所有的意思.

對於二維的陣列有同樣的效果,只是取的方法要考慮的行或列

a=np.eye(5)
>>> b=a[:,:3]
>>> print a
[[1. 0. 0. 0. 0.]
 [0. 1. 0. 0. 0.]
 [0. 0. 1. 0. 0.]
 [0. 0. 0. 1. 0.]
 [0. 0. 0. 0. 1.]]
>>> print b
[[1. 0. 0.]
 [0. 1. 0.]
 [0. 0. 1.]
 [0. 0. 0.]
 [0. 0. 0.]]

又取行又取列

b=a[:3,:3]
>>> print b
[[1. 0. 0.]
 [0. 1. 0.]
 [0. 0. 1.]]
 

以上這篇python [:3] 實現提取陣列中的數就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。