python numpy.linspace
时间: 2023-09-07 12:16:37 浏览: 194
`numpy.linspace` 是一个用于生成等差数列的函数。它的语法如下:
```python
numpy.linspace(start, stop, num=50, endpoint=True, retstep=False, dtype=None, axis=0)
```
参数说明:
- `start`:数列的起始值
- `stop`:数列的结束值
- `num`:要生成的数列的元素个数,默认为 50
- `endpoint`:控制数列是否包含 `stop` 值,默认为 True,即包含 `stop` 值
- `retstep`:是否返回数列的步长,默认为 False
- `dtype`:数列的数据类型,默认为 None,即自动推断
- `axis`:指定数列生成的方向,默认为 0,即沿着第一个轴生成数列
示例代码:
```python
import numpy as np
# 生成从 0 到 1 的等差数列,共 11 个元素
arr = np.linspace(0, 1, num=11)
print(arr)
# 输出:[0. 0.1 0.2 0.3 0.4 0.5 0.6 0.7 0.8 0.9 1. ]
# 生成从 -1 到 1 的等差数列,共 5 个元素
arr = np.linspace(-1, 1, num=5)
print(arr)
# 输出:[-1. -0.5 0. 0.5 1. ]
# 返回数列的步长
arr, step = np.linspace(0, 1, num=5, retstep=True)
print(arr)
# 输出:[0. 0.25 0.5 0.75 1. ]
print(step)
# 输出:0.25
```
希望能帮到你!如果有更多问题,请
阅读全文