Oii Pedro, como você está?
Há várias formas de fazer isso, vou te mostrar duas delas:
np.nan_to_num(km_media)
km_media[np.isnan(km_media)] = 0
Em ambos os exemplos acima a modificação foi feita após a operação, mas é possível fazer diretamente:
km_media = np.nan_to_num(km / (anos - 2019))
# ou
km_media[np.isnan(km / (anos - 2019))] = 0
`
Qual utilizar? Fica a seu critério. No caso da segunda abordagem, se quiséssemos substituir por outro número que não zero, seria possível. Já na primeira, é uma função do próprio Numpy que é responsável por substituir NaN
por zero por padrão, mas é possível alterar esse comportamento utilizando o parâmetro nan
e passando algum valor diferente de zero: np.nan_to_num(km_media, nan=7)
, então, tanto faz qual abordagem utilizar.
Para para consultas futuras sobre a função nan_to_num, deixo como referência a documentação da biblioteca.
Qualquer dúvida fico à disposição.
Abraços e bons estudos!