numpy.geomspace 是 NumPy 库中的一个函数,用于生成对数刻度上的等比例数值序列。这在处理几何级数、对数图等问题时非常有用。
语法
numpy.geomspace(start, stop, num=50, endpoint=True, dtype=None, axis=0)
参数说明
start:序列的起始值(必须为正数)。
stop:序列的终止值(必须为正数)。
num:要生成的等间隔样本数量。默认值是 50。
endpoint:如果为 True(默认值),则 stop 是序列中的最后一个值。如果为 False,则序列中不包含 stop。
dtype:输出数组的数据类型。如果没有提供,则取决于其他参数的类型。
axis:结果中沿指定轴放置样本。默认是 0。
示例
生成从 1 到 1000 的 4 个等比例的数
import numpy as np
arr = np.geomspace(1, 1000, num=4)
print(arr)
不包含终止值
import numpy as np
arr = np.geomspace(1, 1000, num=4, endpoint=False)
print(arr)
指定数据类型
import numpy as np
arr = np.geomspace(1, 1000, num=4, dtype=int)
print(arr)
生成从 1 到 16 的 5 个等比例的数,并设置 dtype 为浮点数
import numpy as np
arr = np.geomspace(1, 16, num=5, dtype=float)
print(arr)