Python 實現(xiàn)取多維數(shù)組第n維的前幾位
更新時間:2019年11月26日 15:52:09 作者:Kenn7
今天小編就為大家分享一篇Python 實現(xiàn)取多維數(shù)組第n維的前幾位,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
現(xiàn)在我們有一個shape為(7352, 9, 128, 1)的numpy數(shù)組。
想要取出第2維的前三個數(shù)據(jù),構(gòu)成新數(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)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
您可能感興趣的文章:
- Python 初始化多維數(shù)組代碼
- python中numpy的矩陣、多維數(shù)組的用法
- python統(tǒng)計多維數(shù)組的行數(shù)和列數(shù)實例
- Python動態(tài)生成多維數(shù)組的方法示例
- Python高級特性——詳解多維數(shù)組切片(Slice)
- python實現(xiàn)在多維數(shù)組中挑選符合條件的全部元素
- 在python中創(chuàng)建指定大小的多維數(shù)組方式
- Python中矩陣創(chuàng)建和矩陣運算方法
- Python矩陣常見運算操作實例總結(jié)
- Python實現(xiàn)的矩陣轉(zhuǎn)置與矩陣相乘運算示例
- python簡單實現(xiàn)矩陣的乘,加,轉(zhuǎn)置和逆運算示例
- Python操作多維數(shù)組輸出和矩陣運算示例
相關(guān)文章
python程序運行添加命令行參數(shù)argparse模塊具體用法詳解
這篇文章主要給大家介紹了關(guān)于python程序運行添加命令行參數(shù)argparse模塊具體用法的相關(guān)資料,argparse是Python內(nèi)置的一個用于命令項選項與參數(shù)解析的模塊,通過在程序中定義好我們需要的參數(shù),需要的朋友可以參考下2024-01-01