Функция NumPy arange() в Python - что это и как использовать?

Что такое функция numpy.arange()?

Функция numpy.arange() в Python создает новый массив numpy с равномерно расположенными числами между началом(включительно) и остановкой(исключительно) с заданным шагом:

numpy.arange(start, stop, step, dtype=None, *, like=None)

Например, в следующем примере используется функция arange() для создания массива numpy:

import numpy as np

a = np.arange(1, 10, 2)

print(a)

Выход:

[1 3 5 7 9]

Массив numpy начинается с 1 и заканчивается на 9. Обратите внимание, что он не включает стоп-значение(10). Поскольку шаг равен 2, массив numpy содержит числа 1, 3, 5, 7 и 9.

Поскольку мы передаем 1 и 10 как целые числа, функция arange() создает новый массив целых чисел. Если вы хотите создать массив чисел с плавающей запятой, вы можете передать начальные и конечные значения как числа с плавающей запятой следующим образом:

import numpy as np

a = np.arange(1.0, 10.0, 2)

print(a)

Выход:

[1. 3. 5. 7. 9.]

Или можете явно указать тип элементов массива numpy, используя аргумент dtype:

import numpy as np

a = np.arange(1, 10, 2, dtype=np.float64)

print(a)

Выход:

[1. 3. 5. 7. 9.]