温馨提示×

numpy.arange怎样生成数组

小樊
91
2024-09-23 12:01:38
栏目: 编程语言

numpy.arange() 函数用于生成等差数列数组。它的基本语法如下:

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

参数说明:

  • start(可选):等差数列的起始值,默认值为 0。
  • stop(必需):等差数列的结束值(不包括该值),必须指定。
  • step(可选):等差数列的公差,默认值为 1。如果未指定,则生成连续整数序列。
  • dtype(可选):生成的数组的元素类型,默认值为 float64

示例:

import numpy as np

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

# 生成一个从 5 到 -5 的等差数列数组,公差为 -1
arr2 = np.arange(5, -5, -1)
print(arr2)  # 输出:[ 5  4  3  2  1  0 -1 -2 -3 -4]

# 生成一个包含 1 到 10 的整数数组,元素类型为 int32
arr3 = np.arange(1, 11, dtype=np.int32)
print(arr3)  # 输出:[1 2 3 4 5 6 7 8 9 10]

0