numpy配列（３次元配列）で特定の値以上を持つ要素を含む行だけ抽出する方法

Question

下の画像のように，3次元のnumpy配列において，赤で囲った列において特定の条件（例えば1300以上）を満たす行を抽出する方法をご教示いただけますとありがたいです．

Answer 1

import numpy as np
from numpy.random import default_rng

rg = default_rng()
x = rg.integers(1000, 1500, 200).reshape((100, 1, 2))

#
y = x[x[:,:,1]>1300].reshape(-1, *x.shape[1:])
print('shape =', y.shape)
print('min =', y[:,:,1].min())
print(y)

#
shape = (43, 1, 2)
min = 1307
[[[1234 1469]]

 [[1359 1353]]

 [[1246 1404]]

 [[1241 1457]]

 [[1401 1388]]

 [[1451 1471]]
       :

Comments

[LenItsuki]

できました．誠にありがとうございます．