温馨提示×

numpy arange函数如何使用

小亿
100
2023-10-09 02:55:59
栏目: 编程语言

numpy的arange函数用于创建一个数组,该数组是按指定步长从起始值到结束值的一系列值。

arange函数的使用方式如下:

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

参数说明:

  • start:起始值,生成的数组中的第一个值,默认为0。

  • stop:结束值,生成的数组中的最后一个值(不包括该值)。

  • step:步长,两个相邻值之间的间隔,默认为1。

  • dtype:生成数组的数据类型,默认为None。

示例代码:

import numpy as np
# 生成一个从0到4的数组
a = np.arange(5)
print(a)  # [0 1 2 3 4]
# 生成一个从2到8的数组,步长为2
b = np.arange(2, 10, 2)
print(b)  # [2 4 6 8]
# 生成一个从1到5的数组,步长为0.5
c = np.arange(1, 5, 0.5)
print(c)  # [1.  1.5 2.  2.5 3.  3.5 4.  4.5]
# 生成一个整型数组
d = np.arange(5, dtype=int)
print(d)  # [0 1 2 3 4]

需要注意的是,arange函数生成的数组不包括结束值,即生成的数组中不包含等于stop的值。如果要包含结束值,可以使用linspace函数。

0