numpy.arange
是NumPy庫中的一個函數,用于生成等差數列。它的基本語法如下:
numpy.arange([start,] stop[, step,], dtype=None)
參數說明:
start
(可選):序列開始的值。如果未提供,則默認為0。stop
(必需):序列結束的值(但不包括該值)。step
(可選):序列中相鄰元素之間的差。如果未提供,則默認為1。dtype
(可選):輸出數組的元素類型。如果未提供,則根據輸入參數自動推斷。以下是一些使用numpy.arange
的示例:
import numpy as np
# 生成一個從0到9的等差數列,步長為1
a = np.arange(10)
print(a) # 輸出:[0 1 2 3 4 5 6 7 8 9]
# 生成一個從0到29的等差數列,步長為10
b = np.arange(0, 30, 10)
print(b) # 輸出:[ 0 10 20]
# 生成一個從5到15的等差數列,步長為2
c = np.arange(5, 16, 2)
print(c) # 輸出:[ 5 7 9 11 13 15]
# 生成一個浮點數等差數列,從0到1,步長為0.1
d = np.arange(0, 1.1, 0.1)
print(d) # 輸出:[ 0. 0.1 0.2 0.3 0.4 0.5 0.6 0.7 0.8 0.9 1. ]
注意:numpy.arange
返回的是一個NumPy數組,因此你可以對其進行其他NumPy操作,如切片、索引等。