溫馨提示×

python中numpy.arange的用法技巧

小樊
81
2024-09-23 12:09:36
欄目: 編程語言

numpy.arange是NumPy庫中的一個函數,用于生成等差數列。它的基本語法如下:

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

參數說明:

  • start(可選):序列的起始值,默認值為0。
  • stop(必需):序列的結束值(不包括該值),必須是整數。
  • step(可選):序列中相鄰元素之間的差值,默認值為1。如果為負數,則生成遞減序列。
  • dtype(可選):輸出數據的類型,默認值為float64??梢灾付槠渌麛抵殿愋?,如int32、float32等。

用法技巧:

  1. 生成0到10之間的整數序列:
import numpy as np
seq = np.arange(11)
print(seq)  # 輸出:[0 1 2 3 4 5 6 7 8 9 10]
  1. 生成5到-5之間的整數序列,步長為2:
import numpy as np
seq = np.arange(5, -5, -2)
print(seq)  # 輸出:[ 5  3 -1 -3 -5]
  1. 生成浮點數序列,起始值為0.0,結束值為1.0,步長為0.1:
import numpy as np
seq = np.arange(0.0, 1.0, 0.1)
print(seq)  # 輸出:[0.  0.1 0.2 0.3 0.4 0.5 0.6 0.7 0.8 0.9]
  1. 生成指定類型的序列:
import numpy as np
seq = np.arange(0, 10, dtype=np.int32)
print(seq)  # 輸出:[0 1 2 3 4 5 6 7 8 9]

0