Casas decimais

Oi, Rafael! Tudo bem com você?

Você pode sim! A biblioteca Numpy nos fornece um método que podemos definir como nosso array será mostrado, esse método é o set_printoptions e com o parâmetro formatter é possível definir a formatação que nosso array será apresentado.

Nesse parâmetro precisamos definir um dicionário que irá ter como chave algum dos valores definidos na documentação que se refere ao que desejamos realizar. Como queremos definir um tamanho para casa decimais em um float precisamos utilizar a chave ‘float_kind’ que irá configurar valores float e o elemento dela deve ser a formatação que desejamos. Montei aqui um exemplo que mostra 3 casas decimais após a virgula com a variável km_media:

np.set_printoptions(formatter={'float_kind':'{:.3f}'.format})
km_media

Os primeiros valores de km_media são mostrados a seguir:

array([2775.625, 204.000, 1280.103, nan, 1981.308, 1532.571, nan, ...

Você pode utilizar dessa estrutura para montar várias outras que desejar para sua aplicação, no link da documentação colocado mais acima, você pode conhecer mais formas de se trabalhar com esse método e também com outros parâmetros ;-)

Eu espero ter te ajudado! Se surgir outra dúvida estarei à disposição.

Bons estudos!