什么是np.linspace函数?
np.linspace函数是NumPy中的一个函数,用于创建等差数列。等差数列是指数列中相邻两项之差相等的数列。np.linspace函数可以指定数列的起始值、终止值和元素个数,从而生成一个等差数列。
np.linspace函数的语法格式
np.linspace函数的语法格式如下:
numpy.linspace(start, stop, num=50, endpoint=True, retstep=False, dtype=None)
- start:数列的起始值。
- stop:数列的终止值。
- num:生成的元素个数,默认为50。
- endpoint:是否包含终止值,默认为True。
- retstep:是否返回元素间的步长,默认为False。
- dtype:生成数列的数据类型,默认为None。
np.linspace函数的使用方法
np.linspace函数的使用方法如下:
- 生成一个包含5个元素的等差数列:
- 生成一个包含10个元素的等差数列,不包含终止值:
- 生成一个包含5个元素的等差数列,并返回元素间的步长:
import numpy as np
a = np.linspace(1, 10, 5)
print(a)
输出结果为:[ 1. 3.25 5.5 7.75 10. ]
b = np.linspace(0, 1, 10, endpoint=False)
print(b)
输出结果为:[0. 0.1 0.2 0.3 0.4 0.5 0.6 0.7 0.8 0.9]
c, step = np.linspace(1, 10, 5, retstep=True)
print(c, step)
输出结果为:[ 1. 3.25 5.5 7.75 10. ] 2.25
