温馨提示×

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。可以指定为其他数值类型,如int32float32等。

用法技巧:

  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