您好,登錄后才能下訂單哦!
現(xiàn)在我們有一個shape為(7352, 9, 128, 1)的numpy數(shù)組。
想要取出第2維的前三個數(shù)據(jù),構成新數(shù)組(7352, 3, 128, 1)
我的思想是:將第2維數(shù)據(jù)轉(zhuǎn)置(transpose)到第一維,再用切片(slice)取出前三個數(shù)據(jù),再轉(zhuǎn)置回去:
print("# original", input.shape) input_transpose = input.transpose((1, 0, 2, 3)) print("# transpose", input_transpose.shape) input_slice = input_transpose[0:3] print("# slice", input_slice.shape) output = input_slice.transpose((1, 0, 2, 3)) print("# output", output.shape)
其實更簡單的做法是:
print("# original", input.shape) print("# output", input[:, [0, 1, 2]].shape)
以上這篇Python 實現(xiàn)取多維數(shù)組第n維的前幾位就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持億速云。
免責聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點不代表本網(wǎng)站立場,如果涉及侵權請聯(lián)系站長郵箱:is@yisu.com進行舉報,并提供相關證據(jù),一經(jīng)查實,將立刻刪除涉嫌侵權內(nèi)容。