我有一个大小为 80x40 的数组,并希望根据特定列 (10) 中的值将每一行发送到两个较小的数组之一。我有类似于下面的代码,但这最终会使数组变平。我不知道输出数组(Array2、Array3)的 Y 维度。我想我可以用一些代码来计算 50 以上和以下的所有值,以获得输出轴的 Y 尺寸,然后制作 2 个 np.zeros(Array.shape[0],Yvalues) 输出数组并逐行 append 到但我仍然不确定这将如何运作。
Array.shape=(80,40)
Array2=[]
Array3=[]
for x in range(0,Array.shape[0]):
if Array[x,10]<50:
Array2.append(Array[x,:])
else:
Array3.append(Array[x,:])
最佳答案
作为一个较小的示例:
a = np.array([[1, 10], [1, 20], [2, 30], [2, 40], [1, 50], [3, 60], [1, 70]])
a2 = a[a[:, 0] < 1.5]
a3 = a[a[:, 0] >= 1.5]
a2
现在是:
array([[ 1, 10],
[ 1, 20],
[ 1, 50],
[ 1, 70]])
和a3
现在是:
array([[ 2, 30],
[ 2, 40],
[ 3, 60]])
因此,根据您的情况,请使用:
a2 = a[a[:, 10] < 50]
a3 = a[a[:, 10] >= 50]
关于python - 在python中将数组拆分为两个较小的数组,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/28599405/