在Python的NumPy庫中,arange函數用于生成一個等差數列。它的定義如下:
numpy.arange([start,] stop[, step,], dtype=None)
參數說明:
start(可選):等差數列的起始值,默認值為0。stop(必需):等差數列的結束值(不包括該值),必須指定。step(可選):等差數列中相鄰兩項之間的差,默認值為1。如果指定了step,則start值會自動調整為start + step。dtype(可選):返回值的類型,默認值為float64??梢灾付槠渌麛抵殿愋?,如int8、int16、float32等。示例:
import numpy as np
# 生成一個從0到9的等差數列,步長為2
arr = np.arange(0, 10, 2)
print(arr) # 輸出:[0 2 4 6 8]
# 生成一個從2到18的等差數列,步長為3
arr2 = np.arange(2, 19, 3)
print(arr2) # 輸出:[ 2 5 8 11 14 17]