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 中执行其他常见操作:

如何用值填充 NumPy 数组
如何替换 NumPy 数组中的元素
如何计算 NumPy 数组中的唯一值

添加评论

您的电子邮箱地址不会被公开。 必填项已用*标注