将 3d numpy 数组分成 2 组 python

Question

我有一个这种形式的 3d 数组 (12457,8,6) 我想将它分成 2 个相等的 numpy 数组，例如 (12457,3,8) 事实上，第一个包含前 3 个波段，第二个包含其余波段：换句话说，我希望我的 array1 包含波段 1、2、3，而我的 array2 包含波段 4、5、6

我试过了，但没用

array1=data[:,:,3]
array1.shape
(12457,8)

Answer 1

您可以使用 np.split -

X = np.random.random((1200,6,8))
print(X.shape)

X1, X2 = np.split(X, 2, axis=1) #Array, num of splits, axis for splitting
print(X1.shape, X2.shape)

(1200, 6, 8)
(1200, 3, 8) (1200, 3, 8)

Answer 2

split or array_split应该能帮到你。

import numpy as np

arr = np.array([[1,2,3], [4, 5, 6], [7, 8, 9], [10, 11, 12]])
newarr = np.split(arr, 2)
print(newarr)

打印：

[array([[1, 2, 3],
       [4, 5, 6]]), array([[ 7,  8,  9],
       [10, 11, 12]])]

Divide a 3d numpy array into 2 groups python