numpy.geomspace

numpy.geomspace是一个函数，用于生成指定起始点和终止点之间等比数列的一维数组，可用于数值计算与数据分析领域。

用法

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

参数

• start: 开始点，必须为正数。
• stop: 终止点，必须为正数。
• num: 数列中的点数。默认为50。
• endpoint: 是否包含终止点。默认为True，即包含终止点。
• dtype: 输出数组的数据类型。如果未提供，则使用数组中提供的现有数据类型。
• axis: 沿着哪个轴生成数组。默认为0，即生成一维数组。

返回值

返回一个一维数组，其中的数值等比分布在 start 和 stop 之间。

示例

生成一维数组

>>> import numpy as np
>>> np.geomspace(1, 100, num=10)
array([  1.        ,   1.66810054,   2.7825594 ,   4.64158883,
         7.74263683,  12.91549665,  21.5443469 ,  35.93813664,
        59.94842503, 100.        ])

在二维数组中生成等比数列

>>> a = np.array([[1, 2], [3, 4]])
>>> np.geomspace(a, 100, endpoint=False)
array([[  1.        ,   1.46779927,   2.15443469,   3.16227766,
          4.64158883,   6.81292069,  10.        ],
       [  3.        ,   4.4011201 ,   6.47137449,   9.49282068,
         13.93213911,  20.43359718,  30.        ]])

总结

numpy.geomspace用于生成以等比间隔排列的数列，可以用于各种计算及可视化中。使用时，需注意参数的取值范围，以避免出错。