np.linspace()函数作用
时间: 2023-11-20 11:57:04 浏览: 120
np.linspace()函数是numpy库中的一个函数,用于在指定的间隔内返回均匀间隔的数字。它的语法如下:
```python
numpy.linspace(start, stop, num=50, endpoint=True, retstep=False, dtype=None, axis=0)
```
其中,参数的含义如下:
- start:序列的起始值。
- stop:序列的结束值。
- num:生成的等间隔样例数量,默认为50。
- endpoint:序列中是否包含stop值,默认为True。
- retstep:如果为True,返回样例之间的间距。
- dtype:输出数组的数据类型。
- axis:沿着哪个轴生成样例,默认为0。
下面是一些np.linspace()函数的例子:
```python
import numpy as np
# 生成一个包含10个元素的等间隔序列,起始值为0,结束值为2
ni1 = np.linspace(0, 2, 10)
print(ni1) # 输出:[0. 0.22222222 0.44444444 0.66666667 0.88888889 1.11111111 1.33333333 1.55555556 1.77777778 2. ]
# 生成一个包含10个元素的等间隔序列,起始值为0,结束值为2,但不包含结束值2
ni2 = np.linspace(0, 2, 10, endpoint=False)
print(ni2) # 输出:[0. 0.2 0.4 0.6 0.8 1. 1.2 1.4 1.6 1.8]
# 生成一个包含10个元素的等间隔序列,起始值为0,结束值为2,返回样例之间的间距
ni3 = np.linspace(0, 2, 10, retstep=True)
print(ni3) # 输出:(array([0. , 0.22222222, 0.44444444, 0.66666667, 0.88888889, 1.11111111, 1.33333333, 1.55555556, 1.77777778, 2. ]), 0.2222222222222222)
# 生成一个包含10个元素的等间隔序列,起始值为0,结束值为2,数据类型为整型
ni4 = np.linspace(0, 2, 10, dtype=int)
print(ni4) # 输出:[0 0 0 0 0 1 1 1 1 2]
```
阅读全文