numpy的arange函數(shù)用于創(chuàng)建一個數(shù)組,該數(shù)組是按指定步長從起始值到結(jié)束值的一系列值。
arange函數(shù)的使用方式如下:
numpy.arange(start, stop, step, dtype=None)
參數(shù)說明:
start:起始值,生成的數(shù)組中的第一個值,默認為0。
stop:結(jié)束值,生成的數(shù)組中的最后一個值(不包括該值)。
step:步長,兩個相鄰值之間的間隔,默認為1。
dtype:生成數(shù)組的數(shù)據(jù)類型,默認為None。
示例代碼:
import numpy as np
# 生成一個從0到4的數(shù)組
a = np.arange(5)
print(a) # [0 1 2 3 4]
# 生成一個從2到8的數(shù)組,步長為2
b = np.arange(2, 10, 2)
print(b) # [2 4 6 8]
# 生成一個從1到5的數(shù)組,步長為0.5
c = np.arange(1, 5, 0.5)
print(c) # [1. 1.5 2. 2.5 3. 3.5 4. 4.5]
# 生成一個整型數(shù)組
d = np.arange(5, dtype=int)
print(d) # [0 1 2 3 4]
需要注意的是,arange函數(shù)生成的數(shù)組不包括結(jié)束值,即生成的數(shù)組中不包含等于stop的值。如果要包含結(jié)束值,可以使用linspace函數(shù)。