python - 是否有一个 numpy 函数可以让你指定开始、步骤和数字？

Question

我们都熟悉 np.linspace，它创建一个给定 start、stop 和 num< 的数组 元素:

In [1]: import numpy as np

In [2]: np.linspace(0, 10, 9)
Out[2]: array([  0.  ,   1.25,   2.5 ,   3.75,   5.  ,   6.25,   7.5 ,   8.75,  10.  ])

同样，谁能忘记 np.arange，它会在给定 start、stop 和 step< 的情况下创建一个数组:

In [4]: np.arange(0, 10, 1.25)
Out[4]: array([ 0.  ,  1.25,  2.5 ,  3.75,  5.  ,  6.25,  7.5 ,  8.75])

但是是否有一个函数允许您指定元素的start、step 和num，同时省略停止？应该有。

Answer 1

谢谢你的提问。我遇到过同样的问题。(从我的角度来看)最短和最优雅的方法是:

import numpy as np
start=0
step=1.25
num=9

result=np.arange(0,num)*step+start

print(result)

返回

[  0.     1.25   2.5    3.75   5.     6.25   7.5    8.75  10.  ]