Python 實現取多維數組第n維的前幾位
更新時間：2019年11月26日 15:52:09 作者：Kenn7
今天小編就為大家分享一篇Python 實現取多維數組第n維的前幾位，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
現在我們有一個shape為(7352, 9, 128, 1)的numpy數組。
想要取出第2維的前三個數據，構成新數組(7352, 3, 128, 1)
我的思想是：將第2維數據轉置（transpose）到第一維，再用切片（slice）取出前三個數據，再轉置回去：
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 實現取多維數組第n維的前幾位就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
python使用celery實現異步任務執(zhí)行的例子
今天小編就為大家分享一篇python使用celery實現異步任務執(zhí)行的例子，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-08-08
成語填空想必大家都是十分熟悉的了，特別是有在上小學的家長肯定都有十分深刻的印象。當然了你也別小看了成語調控小游戲，有的時候知識儲備不夠，你還真的不一定猜得出來是什么。本文就來用Python編寫一個簡單的成語填空游戲，感興趣的可以了解下2023-02-02
各位有沒有遇到過需要批量整理多層文件夾結構里各種類型(docx、excel、ppt)的文件材料的時候,下面這篇文章主要給大家介紹了關于如何使用Python一鍵修改上萬個文件名的相關資料,需要的朋友可以參考下2022-05-05
解決tensorflow/keras時出現數組維度不匹配問題
這篇文章主要介紹了解決tensorflow/keras時出現數組維度不匹配問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-06-06
這篇文章主要為大家詳細介紹了python實現心型照片墻效果，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2021-09-09
在本文里我們給大家整理了關于python制作填詞游戲的具體步驟以及實例代碼，需要的朋友們跟著學習下。2019-05-05
這篇文章主要介紹了詳解pygame捕獲鍵盤事件的兩種方式，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2021-04-04
pytorch之pytorch?hook和關于pytorch?backward過程問題
這篇文章主要介紹了pytorch之pytorch?hook和關于pytorch?backward過程問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-09-09
讀萬卷書不如行萬里路，學的扎不扎實要通過實戰(zhàn)才能看出來，本篇文章手把手帶你爬下肯德基的官網,大家可以在過程中查缺補漏，看看自己掌握程度怎么樣2021-10-10
Python讀取txt文件數據的方法(用于接口自動化參數化數據)
這篇文章主要介紹了Python讀取txt文件數據的方法(用于接口自動化參數化數據),需要的朋友可以參考下2018-06-06