NumPy 教程

NumPy 笔记

numpy linspace 函数的用法详解

numpy 笔记 numpy 笔记


numpy.linspace 函数是仿照 matlab 的 linspace 函数实现的,它返回指定间隔的等间隔数字列表;主要用于创建一个一维数组,并且是等差数列构成的一维数组。

函数

def linspace(start, stop, num=50, endpoint=True, retstep=False, dtype=None, axis=0):

它返回在间隔 [ start,stop ] 内计算的 num 个均匀间隔的样本。

具体参数(Parameters)意义如下:

start : array_like
序列的初始值。
stop : array_like
序列的结束值。
num : int, optional
要生成的样本数,默认值是 50,必须是非负数。
endpoint : bool, optional
表示是否包含结束值,默认是包含,即为 True。
retstep : bool, optional
返回结果是否包含样本间隔值,如果设置为 True,则返回 (samplesstep),默认是 False,即不返回间隔值。
dtype : dtype, optional
输出数组的类型;如果 dtype 未指定,则根据输入的参数自动推断数据类型,该参数 1.9.0 版本开始支持
axis : int, optional
结果中的轴用于存储样本,该参数 1.16.0 版本开始支持

返回值(Returns)意义如下:

samples : ndarray
等间距的 num 个样本数据列表。
step : float, optional
当入参 retstep 为 True 时,返回样本间隔值。

示例

>>> import numpy as np
>>> np.linspace(1.0, 88.0, num=5)
array([ 1.  , 22.75, 44.5 , 66.25, 88.  ])
>>> np.linspace(1.0, 88.0, num=5, endpoint=False)
array([ 1. , 18.4, 35.8, 53.2, 70.6])
>>> np.linspace(1.0, 88.0, num=5, retstep=True)
(array([ 1.  , 22.75, 44.5 , 66.25, 88.  ]), 21.75)