Numpy Array Boolean Indexing, чтобы получить сдерживающий элемент ⇐ Python

[Anonymous]

Для массива (3,2,2) как мне получить элементы второго измерения с заданным единственным значением в третьем измерении
import numpy as np

arr = np.array([
[[31., 1.], [41., 1.]],
[[63., 1.],[73., 3.]],
[[ 95., 1.], [100., 1]]
]
)

ref = arr[(arr[:,:,0] > 41.) & (arr[:,:,0]
Ожидаемый результат < /p>
[[63., 1.],[73., 3.]]

Входное значение — 63, поэтому я заранее не знаю, существует ли 73, но я тоже хочу его вернуть. Другими словами, если значение существует, верните весь родительский массив без изменения формы.
Другой пример
ref = arr[(arr[:,:,0]

Подробнее здесь: https://stackoverflow.com/questions/793 ... ng-element