NumPy: вернуть 0 с делением на ноль

Я пытаюсь выполнить поэлементное деление в python, но если встречается ноль, мне нужно, чтобы коэффициент был просто нулем.

Например:

array1 = np.array([0, 1, 2])
array2 = np.array([0, 1, 1])

array1 / array2 # should be np.array([0, 1, 2])

Я всегда мог просто использовать цикл for для своих данных, но для того, чтобы действительно использовать оптимизацию numpy, мне нужна функция деления, чтобы возвращать 0 при делении на ноль ошибок вместо того, чтобы игнорировать ошибку.

Если я что-то упустил, это не кажетсяnumpy.seterr () может возвращать значения при ошибках. У кого-нибудь есть какие-либо другие предложения о том, как я мог бы извлечь максимальную выгоду из NumPy при настройке собственного деления на ноль обработки ошибок?

Ответы на вопрос(8)

Ваш ответ на вопрос