Estou tentando executar uma divisão sábia do elemento em python, mas se um zero for encontrado, preciso que o quociente seja zero.

Por exemplo:

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

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

Eu sempre poderia usar um loop for nos meus dados, mas para realmente utilizar as otimizações do numpy, preciso da função de divisão para retornar 0 após a divisão por zero de erros em vez de ignorar o erro.

A menos que esteja faltando alguma coisa, não parecenumpy.seterr () pode retornar valores mediante erros. Alguém tem alguma outra sugestão sobre como eu poderia tirar o melhor proveito do numpy enquanto definia minha própria divisão por zero tratamento de erros?