Функция geomspace

Функция geomspace создает массив чисел, распределенных по геометрической прогрессии. Первым параметром передается начальное значение, вторым - конечное. Третьим параметром указывается количество элементов для генерации. По умолчанию параметр num равен 50.

Синтаксис

np.geomspace(start, stop, num=50, endpoint=True, dtype=None, axis=0)

Пример

Создадим массив из 5 чисел, равномерно распределенных по геометрической шкале от 1 до 100:

import numpy as np

arr = np.geomspace(1, 100, num=5)
print(arr)

Результат выполнения кода:

[ 1. 3.16227766 10. 31.6227766 100. ]

Пример

Исключим конечное значение, передав параметр endpoint со значением False:

import numpy as np

arr = np.geomspace(1, 100, num=5, endpoint=False)
print(arr)

Результат выполнения кода:

[ 1. 2.51188643 6.30957344 15.84893192 39.81071706]

Пример

Укажем тип данных элементов с помощью параметра dtype:

import numpy as np

arr = np.geomspace(1, 1000, num=4, dtype=np.int32)
print(arr)

Результат выполнения кода:

[ 1 21 464 1000]

Пример

Создадим массив с отрицательными значениями. Функция корректно работает с отрицательными числами:

import numpy as np

arr = np.geomspace(-10, -1000, num=4)
print(arr)

Результат выполнения кода:

[ -10. -46.41588834 -215.443469 -1000. ]

Пример

Используем geomspace для генерации данных для построения логарифмической шкалы графика:

import numpy as np

x = np.geomspace(0.1, 100, num=10)
y = np.log(x)

print("x:", x)
print("y:", y)

Результат выполнения кода:

x: [  0.1		  0.16681005   0.27825594   0.46415888   0.77426368
	 1.29154967   2.15443469   3.59381366   5.9948425   10.		]
y: [-2.30258509 -1.79004334 -1.27915899 -0.76764843 -0.25562944
	 0.25562805  0.76764794  1.27915116  1.79003485  2.30258509]

Смотрите также

функцию linspace,
которая создает массив чисел с линейным распределением
функцию logspace,
которая создает массив с логарифмической шкалой
функцию arange,
которая создает массив с заданным шагом
функцию fromfunction,
которая создает массив на основе пользовательской функции