快捷導(dǎo)航

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程序運(yùn)行添加命令行參數(shù)argparse模塊具體用法詳解
這篇文章主要給大家介紹了關(guān)于python程序運(yùn)行添加命令行參數(shù)argparse模塊具體用法的相關(guān)資料,argparse是Python內(nèi)置的一個(gè)用于命令項(xiàng)選項(xiàng)與參數(shù)解析的模塊,通過(guò)在程序中定義好我們需要的參數(shù),需要的朋友可以參考下
2024-01-01
Pandas?篩選和刪除目標(biāo)值所在的行的實(shí)現(xiàn)
本文主要介紹了Pandas篩選和刪除目標(biāo)值所在的行的實(shí)現(xiàn)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧
2022-07-07
Python3.8安裝tensorflow的簡(jiǎn)單方法步驟
最近想要研究一下深度學(xué)習(xí)的應(yīng)用,于是翻了翻了社區(qū)的tensorflow安裝教程,所以下面這篇文章主要給大家介紹了關(guān)于Python3.8安裝tensorflow的簡(jiǎn)單方法步驟,需要的朋友可以參考下
2022-06-06
Django跨域請(qǐng)求無(wú)法傳遞Cookie的解決
這篇文章主要介紹了Django跨域請(qǐng)求無(wú)法傳遞Cookie的解決方案，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
2021-04-04
pytorch實(shí)現(xiàn)查看當(dāng)前學(xué)習(xí)率
這篇文章主要介紹了pytorch實(shí)現(xiàn)查看當(dāng)前學(xué)習(xí)率，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
2020-06-06
pandas求兩個(gè)表格不相交的集合方法
今天小編就為大家分享一篇pandas求兩個(gè)表格不相交的集合方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
2018-12-12
python計(jì)算兩個(gè)數(shù)的百分比方法
今天小編就為大家分享一篇python計(jì)算兩個(gè)數(shù)的百分比方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
2018-06-06
簡(jiǎn)單聊聊Python中的鴨子類型和猴子補(bǔ)丁
不知不覺(jué)使用python寫代碼已經(jīng)很長(zhǎng)時(shí)間了,下面這篇文章主要給大家介紹了關(guān)于python鴨子類型(duck?type)和猴子補(bǔ)丁(mokey?patching)的相關(guān)資料,文中通過(guò)示例代碼介紹的非常詳細(xì),需要的朋友可以參考下
2022-04-04
詳解Python中的type()方法的使用
這篇文章主要介紹了詳解Python中的type()方法的使用,是Python入門中的基礎(chǔ)知識(shí),需要的朋友可以參考下
2015-05-05
使用python實(shí)現(xiàn)將視頻中的音頻分離出來(lái)
這篇文章主要介紹了使用python實(shí)現(xiàn)將視頻中的音頻分離出來(lái)，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教
2022-11-11