在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]