numpy三维数组维度变换/提取

有时对多维numpy数组需要进行维度的转换/提取,遇到需要从(A,B,C)三维数组中提取(A,B)、(A,C)或者(B,C)或者(A,)这几个维度数据时,总是忘记该如何切片,记录一下

1. (A,B,C)——> (A,B)

X_New = X[ :,:,0]



在这里插入图片描述

2. (A,B,C)——> (A,C)

X_New = X[ :,0,:]

3. (A,B,C)——> (B,C)

X_New = X[ 0,:,:]

4. (A,B,C)——> (A,)

X_New = X[ :,0,0]

猜你喜欢

转载自blog.csdn.net/MAINKEYS/article/details/123834116