如何使用Numpy生成包含n个等步长元素的数组？

直接用Numpy的linspace函数就能满足需求，它专门用于生成指定区间内固定数量的等间隔数值序列，用法简单直接：

基础用法 #

import numpy as np

# 定义区间[a, b]和元素个数n
a = 2
b = 10
n = 5

# 生成包含n个元素的等步长数组，默认包含区间终点b
result = np.linspace(start=a, stop=b, num=n)
print(result)
# 输出：[ 2.  4.  6.  8. 10.]

参数说明 #

• start：必填，区间的起始值（对应你的a）
• stop：必填，区间的结束值（对应你的b）
• num：必填，要生成的元素总个数（对应你的n）
• endpoint：可选布尔值，默认True，表示序列包含stop值；若设为False，序列会排除stop，此时步长为(b-a)/n而非默认的(b-a)/(n-1)

比如设置endpoint=False的示例：

result_no_end = np.linspace(start=a, stop=b, num=n, endpoint=False)
print(result_no_end)
# 输出：[2.  3.6 5.2 6.8 8.4]

替代方案（用arange） #

如果你想通过计算步长来生成，也可以用arange，但需要先手动计算步长(b-a)/(n-1)，同时注意浮点数精度问题（避免因精度丢失少生成一个元素）：

step = (b - a) / (n - 1)
result_arange = np.arange(a, b + step, step)
print(result_arange)
# 输出：[ 2.  4.  6.  8. 10.]

不过相比之下，linspace更可靠，不需要手动处理步长和精度问题，是这类需求的首选工具。

内容的提问来源于stack exchange，提问作者eresque-