numpy的arange和linspace的用法及其区别

1. arange的用法

numpy.arange()是一个用于生成数组的函数,它可以在给定的区间内返回均匀间隔的值。它的语法是

numpy.arange(start, stop, step, dtype = None)

其中:
start是起始值,
stop是终止值(不包括),
step是步长,
dtype是数据类型。

举例说明:

>>> import numpy as np

# 创建一个从 0 到 4 的一维数组
>>> a = np.arange(5)
>>> print(a)  
[0 1 2 3 4]

# 创建一个从 3 到 9 的一维数组,步长为 2
>>> b = np.arange(3, 10, 2)
>>> print(b)
>[3 5 7 9]

# 创建一个从 1 到 2 的一维数组,步长为 0.1
>>> c = np.arange(1, 2, 0.1)
>>> print(c)
>[1.  1.1 1.2 1.3 1.4 1.5 1.6 1.7 1.8 1.9]

2. linspace的用法

numpy.linspace是一个用来在指定区间内生成等间距的数字的函数。它的基本语法是:

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

其中:
start: 区间的起始值
stop: 区间的终止值
num: 生成的数字个数,默认为50
endpoint: 是否包含终止值,默认为True
retstep: 是否返回步长,默认为False
dtype: 输出数组的数据类型,默认为None,根据输入推断
axis: 输出数组沿着哪个轴排列,默认为0,即第一个轴

下面是一个使用 numpy.linspace 函数的示例:

>>> import numpy as np
>
# 生成一个长度为5的等间隔数组,范围是0到1
>>> x = np.linspace(0, 1, 5)

>>> print(x)
array([0.  , 0.25, 0.5 , 0.75, 1.  ])

3. arange和linspace的区别

numpyarangelinspace都是用来生成等差数列的函数,但是有一些区别:

  • arange是按照给定的起始值、终止值和步长来生成数列,而linspace是按照给定的起始值、终止值和元素个数来生成数列。
  • arange的终止值不一定包含在数列中,而linspace的终止值一定包含在数列中。
  • arange可以生成整数或浮点数类型的数组,而linspace只能生成浮点数类型的数组。

举例说明:

# 使用arange生成从0到10(不包含)以2为步长的数组
>>> import numpy as np
>>> np.arange(0, 10, 2)
array([0, 2, 4, 6, 8])

# 使用linspace生成从0到10(包含)以5个元素为个数的数组
>>> import numpy as np
>>> np.linspace(0, 10, 5)
array([ 0. ,  2.5,  5. ,  7.5, 10. ])

猜你喜欢

转载自blog.csdn.net/weixin_39509073/article/details/129581129