python - 使用 NumPy 生成几何序列

Question

我正在尝试生成一个几何序列，类似于 1、2、4、8...

我有以下代码:

import numpy as np

lower_price = 1
upper_price = 2
total_grids = 10

grid_box = np.linspace(lower_price , upper_price, total_grids, retstep=True)

print(grid_box)

输出:

(array([1.        , 1.11111111, 1.22222222, 1.33333333, 1.44444444,
       1.55555556, 1.66666667, 1.77777778, 1.88888889, 2.        ]), 0.1111111111111111)

此代码创建算术序列，而不是几何序列。我如何修复此代码以生成后者而不是前者？

Answer 1

您正在寻找np.logspace ，而不是np.linspace:

例如，

# Lower bound is 2**0 == 1
# Upper bound is 2**10 == 1024
np.logspace(0, 10, 10, base=2)

输出:

[1.00000000e+00 2.16011948e+00 4.66611616e+00 1.00793684e+01
 2.17726400e+01 4.70315038e+01 1.01593667e+02 2.19454460e+02
 4.74047853e+02 1.02400000e+03]

---

如果您尝试获取 1 到 2 之间的 10 个值，请使用:

# Lower bound is 2**0 == 1
# Upper bound is 2**1 == 2
np.logspace(0, 1, 10, base=2)