pandas - 使用百分比增量创建的 numpy 范围

Question

我想创建一个从 start 到 stop 的 numpy 范围，使用基于先前数字的(固定)百分比增量，例如 start=100 increment=2% to得到 100、102、104.04、106.12 等

显然，这可以通过遍历列表或更优雅地使用生成器来完成:

def pct_incrementer(start, stop, step_pct):
    val = start
    yield val
    while True:
        val += val * step_pct
        if val >= stop:
            break
        yield val

 np.array(list(pct_incrementer(100, 200, 0.02)))
 # or (as I've just learned!)
 np.fromiter(pct_incrementer(100, 200, 0.02), float)  

numpy(或 pandas)是否具有本地执行此操作的功能？ (因为我找不到!)。有没有一种方法可以操纵 np.array([100] * 10) 来做同样的事情？

谢谢!

Answer 1

您可以使用full 和cumprod:

xs = 100 * np.full(np.log(2.0)/np.log(1.02),1.02).cumprod()