NumPy : Comment utiliser arange et inclure un point de terminaison

La fonction NumPy arange peut être utilisée pour créer une séquence de valeurs.

Par défaut, cette fonction n’inclut pas le point de terminaison dans la séquence de valeurs.

Il existe deux façons de contourner ce problème :

Méthode 1 : ajouter la taille du pas au point de terminaison

np.arange(start, stop + step, step)

Méthode 2 : utilisez plutôt la fonction linspace

np.linspace(start, stop, num)

Les exemples suivants montrent comment utiliser chaque méthode dans la pratique.

Exemple 1 : ajouter une taille de pas au point de terminaison

Supposons que nous souhaitions créer une séquence de valeurs allant de 0 à 50 avec un pas de 5.

Si nous utilisons la fonction NumPy arange , le point final de 50 ne sera pas inclus dans la séquence par défaut :

import numpy as np

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

#create array
np.arange(start, stop, step)

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

Pour inclure le point final de 50, nous pouvons simplement ajouter la taille du pas à l’argument stop :

import numpy as np

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

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

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

Notez que le point final de 50 est désormais inclus dans la séquence de valeurs.

Remarque : Vous pouvez trouver la documentation complète de la fonction NumPy arange() ici .

Exemple 2 : utilisez plutôt la fonction linspace

Une autre façon de créer une séquence de valeurs et d’inclure le point de terminaison consiste à utiliser la fonction linspace NumPy, qui inclut le point de terminaison par défaut.

Le code suivant montre comment utiliser cette fonction pour créer une séquence de valeurs allant de 0 à 50 :

import numpy as np

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

#create array
np.linspace(start, stop, num)

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

Notez que le point final de 50 est inclus dans la séquence de valeurs par défaut.

Remarque : Vous pouvez trouver la documentation complète de la fonction NumPy arange() ici .

Ressources additionnelles

Les didacticiels suivants expliquent comment effectuer d’autres opérations courantes dans NumPy :

Comment remplir un tableau NumPy avec des valeurs
Comment remplacer des éléments dans un tableau NumPy
Comment compter les valeurs uniques dans le tableau NumPy