從Numpy 數(shù)組通過索引獲取元素或切片

a=np.arange(20).reshape([4,5])
>>> a
array([[ 0,  1,  2,  3,  4],

       [ 5,  6,  7,  8,  9],

       [10, 11, 12, 13, 14],

       [15, 16, 17, 18, 19]])
>>> a[2,3]    #取一個標量元素
13
>>> a[[2]]    #取第三行，得到一個二維numpy數(shù)組，只有一個元素（一維數(shù)組）

array([[10, 11, 12, 13, 14]])
>>> a[:,[2]]   #取第三列

array([[ 2],

       [ 7],

       [12],

       [17]])
>>> a[1:3,[2,3]]   #取連續(xù)幾行的選定列

array([[ 7,  8],

       [12, 13]])
>>> a[[1,3],[2,3]]         #要同時取選定行，選定列結(jié)果會和理解的不一樣

array([ 7, 18])
>>> b=a[[1,3]]          #正確的方法是先取行，在取列
>>> b

array([[ 5,  6,  7,  8,  9],

       [15, 16, 17, 18, 19]])

>>> b[:,[2,3]]

array([[ 7,  8],

       [17, 18]]
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成，瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明：文章內(nèi)容（如有圖片或視頻亦包括在內(nèi)）由作者上傳并發(fā)布，文章內(nèi)容僅代表作者本人觀點，簡書系信息發(fā)布平臺，僅提供信息存儲服務。

友情鏈接更多精彩內(nèi)容