numpy中arange函數(shù)的用法是什么

小億
104
2024-01-10 10:46:36
欄目: 編程語言

numpy中的arange函數(shù)用于創(chuàng)建一個(gè)等差數(shù)組，它的語法如下：

numpy.arange(start, stop, step, dtype)

參數(shù)說明：

  • start：起始值，默認(rèn)為0。
  • stop：終止值（不包含），必須指定。
  • step：步長，默認(rèn)為1。
  • dtype：返回?cái)?shù)組的數(shù)據(jù)類型，默認(rèn)為None。

該函數(shù)返回一個(gè)一維的numpy數(shù)組，包含從起始值到終止值（不包含）之間，以指定的步長遞增的元素。

例如，可以使用arange函數(shù)創(chuàng)建一個(gè)從0到9的整數(shù)數(shù)組：

import numpy as np

arr = np.arange(10)
print(arr)

輸出結(jié)果為：

[0 1 2 3 4 5 6 7 8 9]

還可以指定起始值、終止值和步長來創(chuàng)建不同的數(shù)組。例如，創(chuàng)建一個(gè)從1到9的數(shù)組，步長為2：

import numpy as np

arr = np.arange(1, 10, 2)
print(arr)

輸出結(jié)果為：

[1 3 5 7 9]

0 踩