温馨提示×

python numpy.arange如何使用

小樊
81
2024-09-23 12:00:39
栏目: 编程语言

numpy.arange()是NumPy库中的一个函数,用于生成等差数列。它的基本语法如下:

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

参数说明:

  • start(可选):序列的起始值,默认值为0。
  • stop(必需):序列的结束值(不包括该值),必须指定。
  • step(可选):序列中相邻元素之间的差值,默认值为1。如果指定了step,则start值会自动调整为start + step
  • dtype(可选):序列中元素的类型,默认值为float64。可以指定为intfloat等类型。

使用示例:

import numpy as np

# 生成一个从0到9的等差数列,步长为2
arr1 = np.arange(0, 10, 2)
print(arr1)  # 输出:[0 2 4 6 8]

# 生成一个从2到18的等差数列,步长为3
arr2 = np.arange(2, 19, 3)
print(arr2)  # 输出:[ 2  5  8 11 14 17]

# 生成一个从5开始,到50之前(不包括50)的等差数列,步长为5
arr3 = np.arange(5, 50, 5)
print(arr3)  # 输出:[ 5 10 15 20 25 30 35 40 45]

# 生成一个浮点数等差数列,从0到1,步长为0.1
arr4 = np.arange(0, 1, 0.1)
print(arr4)  # 输出:[0.  0.1 0.2 0.3 0.4 0.5 0.6 0.7 0.8 0.9]

0