Python 實(shí)現(xiàn)取多維數(shù)組第n維的前幾位

更新時(shí)間：2019年11月26日 15:52:09 作者：Kenn7

今天小編就為大家分享一篇Python 實(shí)現(xiàn)取多維數(shù)組第n維的前幾位，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

現(xiàn)在我們有一個(gè)shape為(7352, 9, 128, 1)的numpy數(shù)組。

想要取出第2維的前三個(gè)數(shù)據(jù)，構(gòu)成新數(shù)組(7352, 3, 128, 1)

我的思想是：將第2維數(shù)據(jù)轉(zhuǎn)置（transpose）到第一維，再用切片（slice）取出前三個(gè)數(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)

其實(shí)更簡(jiǎn)單的做法是：

print("# original", input.shape)
print("# output", input[:, [0, 1, 2]].shape)

以上這篇Python 實(shí)現(xiàn)取多維數(shù)組第n維的前幾位就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

python使用celery實(shí)現(xiàn)異步任務(wù)執(zhí)行的例子
今天小編就為大家分享一篇python使用celery實(shí)現(xiàn)異步任務(wù)執(zhí)行的例子，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
2019-08-08
基于Python實(shí)現(xiàn)成語(yǔ)填空游戲的示例代碼
成語(yǔ)填空想必大家都是十分熟悉的了，特別是有在上小學(xué)的家長(zhǎng)肯定都有十分深刻的印象。當(dāng)然了你也別小看了成語(yǔ)調(diào)控小游戲，有的時(shí)候知識(shí)儲(chǔ)備不夠，你還真的不一定猜得出來(lái)是什么。本文就來(lái)用Python編寫(xiě)一個(gè)簡(jiǎn)單的成語(yǔ)填空游戲，感興趣的可以了解下
2023-02-02
如何使用Python一鍵修改上萬(wàn)個(gè)文件名
各位有沒(méi)有遇到過(guò)需要批量整理多層文件夾結(jié)構(gòu)里各種類型(docx、excel、ppt)的文件材料的時(shí)候,下面這篇文章主要給大家介紹了關(guān)于如何使用Python一鍵修改上萬(wàn)個(gè)文件名的相關(guān)資料,需要的朋友可以參考下
2022-05-05
解決tensorflow/keras時(shí)出現(xiàn)數(shù)組維度不匹配問(wèn)題
這篇文章主要介紹了解決tensorflow/keras時(shí)出現(xiàn)數(shù)組維度不匹配問(wèn)題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
2020-06-06
python實(shí)現(xiàn)心型照片墻效果
這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)心型照片墻效果，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下
2021-09-09
python制作填詞游戲步驟詳解
在本文里我們給大家整理了關(guān)于python制作填詞游戲的具體步驟以及實(shí)例代碼，需要的朋友們跟著學(xué)習(xí)下。
2019-05-05
詳解pygame捕獲鍵盤事件的兩種方式
這篇文章主要介紹了詳解pygame捕獲鍵盤事件的兩種方式，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧
2021-04-04
pytorch之pytorch?hook和關(guān)于pytorch?backward過(guò)程問(wèn)題
這篇文章主要介紹了pytorch之pytorch?hook和關(guān)于pytorch?backward過(guò)程問(wèn)題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教
2023-09-09
Python實(shí)戰(zhàn)練習(xí)之終于對(duì)肯德基下手
讀萬(wàn)卷書(shū)不如行萬(wàn)里路，學(xué)的扎不扎實(shí)要通過(guò)實(shí)戰(zhàn)才能看出來(lái)，本篇文章手把手帶你爬下肯德基的官網(wǎng),大家可以在過(guò)程中查缺補(bǔ)漏，看看自己掌握程度怎么樣
2021-10-10
Python讀取txt文件數(shù)據(jù)的方法(用于接口自動(dòng)化參數(shù)化數(shù)據(jù))
這篇文章主要介紹了Python讀取txt文件數(shù)據(jù)的方法(用于接口自動(dòng)化參數(shù)化數(shù)據(jù)),需要的朋友可以參考下
2018-06-06