Удалить бесконечные значения из массива numpy

Question

Я хотел бы удалить элементы из одного массива B, которые имеют тот же индекс, что и элементы inf, из другого массива A. У меня есть два numpy массива, таких как

A = np.array([1,2,3,4, float('inf')])
B = np.array([5, 6, 7, 8, 9])

Если я сделаю B[A>2], вывод будет array([7, 8, 9]). Однако, если я сделаю B[math.isfinite(A)], то получу ошибку

TypeError: only size-1 arrays can be converted to Python scalars

Как выбрать элементы из B, если значение в A не бесконечно?

Answer 1

Я думаю, у вас есть ответ на ваш вопрос:

B = B[A!= float('inf')]