Python中的np.linspace()

介绍

np.linspace(start, stop, num=50, endpoint=True, retstep=False, dtype=None, axis=0)是Numpy库中的一个函数，用于生成一个等差数列。该函数返回num个在start和stop之间的均匀分布的数值。

参数解释

1. start：序列的起始值。必填参数。

2. stop： 序列的结束值。必填参数。

3. num：要生成的等分点的个数，默认为50个。可选参数。

4. endpoint：序列中是否包含stop值。默认为True，即包含。可选参数。

5. retstep：如果为True，则返回两个参数，第一个为数组，第二个为两个相邻元素之间的间隔大小。默认为False。可选参数。

6. dtype：所需的数组元素类型。默认为None，即根据输入数据自动确定数据类型。可选参数。

7. axis：指定要生成的数组的轴。默认为0，即生成的数组是一维数组。可选参数。

返回值

np.linspace()返回一个一维的ndarray数组。

例子

import numpy as np

# 在 1 和 10 之间生成5个等分数
a = np.linspace(1, 10, num=5)

print(a)
# 输出 [ 1.    3.25  5.5   7.75 10.  ]

# 在 0 和 2π 之间生成50个等分数
b = np.linspace(0, 2*np.pi, 50)

print(b)
# 输出数组包含50个0到2π之间的均匀分布的浮点数

# 使用 retstep 参数来获取相邻两个数之间的间隔大小
c, step = np.linspace(0, 10, num=5, retstep=True)

print(f'c: {c}')
# 输出 [ 0.   2.5  5.   7.5 10. ]

print(f'step: {step}')
# 输出 2.5