Numpy: como usar organizar e incluir um endpoint

A função NumPy arange pode ser usada para criar uma sequência de valores.

Por padrão, esta função não inclui o ponto final na sequência de valores.

Existem duas maneiras de contornar esse problema:

Método 1: adicionar tamanho da etapa ao endpoint

 n.p. arange (start, stop + step, step)

Método 2: use a função linspace

 n.p. linspace (start, stop, num)

Os exemplos a seguir mostram como usar cada método na prática.

Exemplo 1: Adicione um tamanho de passo ao endpoint

Digamos que queremos criar uma sequência de valores de 0 a 50 com passo de 5.

Se usarmos a função NumPy arange , o ponto final de 50 não será incluído na sequência por padrão:

 import numpy as np

#specify start, stop, and step size
start = 0
stop = 50
step = 5

#create array
n.p. arange (start, stop, step)

array([ 0, 5, 10, 15, 20, 25, 30, 35, 40, 45])

Para incluir o ponto final de 50, podemos simplesmente adicionar o tamanho do passo ao argumento stop :

 import numpy as np

#specify start, stop, and step size
start = 0
stop = 50
step = 5

#create array
n.p. arange (start, stop + step, step)

array([ 0, 5, 10, 15, 20, 25, 30, 35, 40, 45, 50])

Observe que o ponto final de 50 agora está incluído na sequência de valores.

Nota : Você pode encontrar a documentação completa para a função NumPy arange() aqui .

Exemplo 2: use a função linspace

Outra forma de criar uma sequência de valores e incluir o endpoint é usar a função NumPy linspace , que inclui o endpoint por padrão.

O código a seguir mostra como usar esta função para criar uma sequência de valores de 0 a 50:

 import numpy as np

#specify start, stop, and number of total values in sequence
start = 0
stop = 50
num = 11

#create array
n.p. linspace (start, stop, num)

array([ 0., 5., 10., 15., 20., 25., 30., 35., 40., 45., 50.])

Observe que o ponto final de 50 está incluído na sequência de valores padrão.

Nota : Você pode encontrar a documentação completa para a função NumPy arange() aqui .

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras operações comuns no NumPy:

Como preencher um array NumPy com valores
Como substituir elementos em um array NumPy
Como contar valores únicos no array NumPy