Как использовать numpy.where в нешироковещательных массивах? ⇐ Python

[Anonymous]

Предположим, у меня есть четыре массива a,b,c и d, где a,

Код: Выделить всё

b

и c имеют одинаковую форму (n,), но форма d может быть совершенно разной. Я хотел бы сделать что-то вроде

Код: Выделить всё

np.where(a < d, b, c)

и хотел бы, чтобы это дало мне массив e формы (n,)+d.shape такой, что e[i,...] = b if d[...] > a и d[i,...] = c if d[...]

Подробнее здесь: https://stackoverflow.com/questions/798 ... ble-arrays