numpy 里面生成等差数列
时间: 2023-10-06 20:12:28 浏览: 109
等差数列.py
你可以使用NumPy中的`arange`函数来生成等差数列。`arange`函数的用法如下:
```python
numpy.arange([start, ]stop, [step, ]dtype=None)
```
其中,`start`表示数列的起始值(默认为0),`stop`表示数列的结束值,`step`表示数列中相邻两个数的差(默认为1),`dtype`表示数列的数据类型(可选参数,如果未指定,则根据输入参数来推断数据类型)。
以下是使用`arange`函数生成等差数列的示例代码:
```python
import numpy as np
# 生成从0到9的等差数列,默认步长为1
arr1 = np.arange(10)
print(arr1) # 输出: [0 1 2 3 4 5 6 7 8 9]
# 生成从2到10的等差数列,步长为2
arr2 = np.arange(2, 11, 2)
print(arr2) # 输出: [ 2 4 6 8 10]
```
通过调整`start`、`stop`和`step`参数的值,你可以根据自己的需求生成不同范围和步长的等差数列。
阅读全文