np.linspace用法
时间: 2023-07-18 14:43:02 浏览: 72
`np.linspace` 是 NumPy 库中的一个函数,用于在指定的起始值和终止值之间生成等间隔的一维数组。
该函数的完整语法如下:
```python
np.linspace(start, stop, num=50, endpoint=True, retstep=False, dtype=None, axis=0)
```
参数解释:
- `start`:起始值
- `stop`:终止值
- `num`:要生成的等间隔样本数量,默认为 50
- `endpoint`:如果为 true,终止值也包含在输出数组之中。默认为 True
- `retstep`:如果为 True,返回值为 (samples, step),其中 step 是样本之间的步长
- `dtype`:输出数组的数据类型
- `axis`:数组的轴。默认为 0,表示输出一维数组
示例:
```python
import numpy as np
# 生成一个长度为10的一维数组,起始值为0,终止值为1
arr1 = np.linspace(0, 1, 10)
print(arr1)
# 生成一个长度为5的一维数组,起始值为1,终止值为10
arr2 = np.linspace(1, 10, 5)
print(arr2)
# 生成一个长度为3的一维数组,起始值为0,终止值为1(不包含终止值)
arr3 = np.linspace(0, 1, 3, endpoint=False)
print(arr3)
```
输出:
```
[0. 0.11111111 0.22222222 0.33333333 0.44444444 0.55555556
0.66666667 0.77777778 0.88888889 1. ]
[ 1. 3.25 5.5 7.75 10. ]
[0. 0.5 1. ]
```
阅读全文