Python 實現取多維陣列第n維的前幾位

2022-10-04 15:42:28 字數 666 閱讀 5454

現在我們有qqamsi乙個shape為(7352, 9, 128, 1)的numpy陣列。

想要取出第2維的前三個資料,構成新陣列(7352, 3, 128, 1)

我的思想是:將第2維資料轉置(transpose)到第一維,再用切片(slice)取出前三個資料,再轉置回去:

print("# original", input.shape)

input_transpose = input. 0, 2, 3))

print("# transpose", input_transpose.shape)

input_slice = input_transpose[0:3]

print("# slice", input_slice.shape)

output = input_slice.tqqamsiranspose((1, 0, 2, 3))

print("# output", output.shape)

其實更簡單的做法是:

print("qqamsi# original", input.shape)

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

本文標題: python 實現取多維陣列第n維的前幾位

本文位址: /jiaoben/python/288941.html

python 實現多維陣列 array 排序

關於多維陣列如何復合排序 如陣列 import numpy a程式設計客棧s np data np.array 2,2,5 2,1,3 1,2,3 3,1,4 data array 2,2,5 2,1,3 1,2,3 3,1,4 將陣列先按照第一列公升序,第二列公升序,第三列公升序的方式排序 ide...

python使用numpy實現多維陣列的排序

1.問題描述 為實現對多維陣列按多列排序,如有三維陣列座標點 x,y,z 現需使其按照x進行公升序排序,其後使y進行公升序排序,最後為z。有4行3列陣列如下 arr 6 3 3 3 1 1 6 2 8 4 0 3 欲使其按如下排序 arrsort 3 1 1 4 0 3 6 2 8 6 3 3 2....

Python 將多維陣列中大於N的資料替換成其他數

import numpy as np import imageio import time 小於8的數等於2 資料大小是3653 2705的二維矩陣 grid imageio.imread wp2001.tif str time time.time 最笨的方式 多重迴圈 耗時 129.2551689...