np.arange()
函数返回一个有起点和终点的固定步长的排列,如[1,2,3,4,5]
,起点是1,终点是6,步长为1
- 只有一个参数时表示
[0, n)
,默认步长为1
np.arange(3)
[0 1 2]
- 有两个参数表示起点到终点,默认步长为1
np.arange(5, 10)
[5 6 7 8 9]
- 三个参数表示起点、终点、步长
np.arange(1, 10, 2)
[1 3 5 7 9]
- 支持小数步长
np.arange(1, 5, 0.5)
[1. 1.5 2. 2.5 3. 3.5 4. 4.5]
np.linspace()
返回start
到stop
之间等间隔的num
个点
numpy.linspace(start, stop, num=50, endpoint=True, retstep=False, dtype=None, axis=0)
np.linspace(2.0, 3.0, num=5)
[2. 2.25 2.5 2.75 3. ]
实例1
x = np.arange(-10, 10, 0.2)
y = np.arange(-10, 10, 0.2)
f_x_y = np.power(x, 2) + np.power(y, 2)
plt.figure()
ax = plt.gca(projection='3d')
ax.plot(x, y, f_x_y)
plt.show()