Numpy:如何使用排列和包含端点
NumPy arange函数可用于创建值序列。
默认情况下,此函数不包括值序列中的端点。
有两种方法可以解决此问题:
方法一:给端点添加步长
n.p. arange (start, stop + step, step)
方法2:使用linspace函数代替
n.p. linspace (start, stop, num)
以下示例展示了如何在实践中使用每种方法。
示例 1:为端点添加步长
假设我们要创建一个从 0 到 50 的值序列,步长为 5。
如果我们使用 NumPy arange函数,默认情况下,端点 50 将不会包含在序列中:
import numpy as np
#specify start, stop, and step size
start = 0
stop = 50
step = 5
#create array
n.p. arange (start, stop, step)
array([ 0, 5, 10, 15, 20, 25, 30, 35, 40, 45])
要包含终点 50,我们可以简单地将步长添加到stop参数:
import numpy as np
#specify start, stop, and step size
start = 0
stop = 50
step = 5
#create array
n.p. arange (start, stop + step, step)
array([ 0, 5, 10, 15, 20, 25, 30, 35, 40, 45, 50])
请注意,端点 50 现在包含在值序列中。
注意:您可以在此处找到 NumPy arange()函数的完整文档。
示例 2:改用 linspace 函数
创建值序列并包含端点的另一种方法是使用 NumPy linspace函数,该函数默认包含端点。
以下代码展示了如何使用该函数创建从 0 到 50 的值序列:
import numpy as np
#specify start, stop, and number of total values in sequence
start = 0
stop = 50
num = 11
#create array
n.p. linspace (start, stop, num)
array([ 0., 5., 10., 15., 20., 25., 30., 35., 40., 45., 50.])
请注意,端点 50 包含在默认值序列中。
注意:您可以在此处找到 NumPy arange()函数的完整文档。
其他资源
以下教程解释了如何在 NumPy 中执行其他常见操作: