np.linspace 的用法及参数说明
np.linspace 是 NumPy 中的一个函数,用于在指定区间内生成均匀分布的数值序列。其基本语法如下:
numpy.linspace(start, stop, num=50, endpoint=True, retstep=False, dtype=None, axis=0)
参数说明:
start:序列的起始值。
stop:序列的结束值。如果 endpoint 设置为 True,则包括该值;如果设置为 False,则排除该值。
num=50:生成的点数(默认是 50 个点)。
endpoint=True:是否包括 stop 值。默认为 True(包含)。
retstep=False:如果设置为 True,返回元组 (values, step),其中 step 是步长。
dtype=None:生成的数组的 dtype 类型。如果为 None,则使用浮点数类型。
axis=0:如果使用多维数组,在该轴上构建值。
返回值:
返回一个 NumPy ndarray,包含指定区间内按步长生成的值。
示例
示例 1:生成 0 到 10 的 5 个均匀分布的数
生成从 0 到 10 的 5 个均匀分布的数
import numpy as np
numbers = np.linspace(0, 10, 5)
print(numbers)
输出:
[ 0. 2.5 5. 7.5 10. ]
示例 2:不包含结束值
numbers = np.linspace(0, 10, 5, endpoint=False)
print(numbers)
输出:
[ 0. 2. 4. 6. 8. ]
示例 3:返回步长
values, step = np.linspace(0, 10, 5, retstep=True)
print("Values:", values)
print("Step:", step)
输出:
Values: [ 0. 2.5 5. 7.5 10. ]
Step: 2.5
示例 4:指定 dtype
numbers = np.linspace(0, 1, 3, dtype=int)
print(numbers)
输出:
[0 0 1]
np.linspace 非常适合用于生成等间隔的数轴数据,常用于绘图和数值运算中。