python 给定起点、终点和数量,获得等差数列,要求每个数取整
时间: 2024-03-27 19:38:44 浏览: 141
python 实现递归求等差数列
5星 · 资源好评率100%
可以使用numpy的`linspace`函数生成等差数列后,再将其取整。下面是一个示例代码,假设起点为1,终点为10,需要获得10个整数,则可以如下生成等差数列并取整:
```
import numpy as np
start = 1
end = 10
num = 10
arr = np.linspace(start, end, num)
arr = np.round(arr).astype(int)
print(arr)
```
输出结果为:
```
[ 1 2 3 4 5 5 6 7 8 10]
```
其中`np.round`函数用于将浮点数四舍五入取整,`astype(int)`用于将数据类型转换为整型。如果需要向下取整或向上取整,则可以使用`np.floor`和`np.ceil`函数。
阅读全文