欢迎访问宙启技术站
智能推送

arange()生成一维数组?

发布时间:2023-07-02 18:34:53

arange()函数是numpy库中的一个函数,用于生成一个一维数组。它的一般形式为np.arange(start, stop, step),其中start指定数组的起始值,stop指定数组的结束值,step指定数组中元素的间隔。

arange()函数可以用于生成一个等差数列,具有很高的灵活性。下面是一些示例:

1. 生成一个从0到9的一维数组:

import numpy as np

arr = np.arange(10)
print(arr)  # [0 1 2 3 4 5 6 7 8 9]

2. 生成一个从1到10的一维数组,步长为2:

import numpy as np

arr = np.arange(1, 11, 2)
print(arr)  # [1 3 5 7 9]

3. 生成一个从10到1的一维数组,步长为-1,可以用于逆序排列:

import numpy as np

arr = np.arange(10, 0, -1)
print(arr)  # [10 9 8 7 6 5 4 3 2 1]

4. 生成一个一维数组,包含小数:

import numpy as np

arr = np.arange(0, 1, 0.1)
print(arr)  # [0.  0.1 0.2 0.3 0.4 0.5 0.6 0.7 0.8 0.9]

5. 使用arange()生成一个等差数列,并利用reshape()函数将其转换为二维数组:

import numpy as np

arr = np.arange(1, 10).reshape(3, 3)
print(arr)
# [[1 2 3]
#  [4 5 6]
#  [7 8 9]]

这些示例展示了arange()函数在生成一维数组方面的应用。利用它可以轻松生成符合特定规律的一维数组,并应用于各种数值计算、数据处理和科学计算任务。