如何在Python中初始化具有一定间距的整数列表

2024-10-01 00:28:38 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试将我的一些Java算法代码翻译成Python

对于以下Java代码,我无法找到一种干净的转换方法:

int[] groupBaseIndex = IntStream.iterate(0, n -> n < size, n -> n + step).toArray();

基本上,它使用step0size生成一个数组。例如,如果size=14step=4,它将生成:

[0,4,8,12]

有人能教我如何将它干净地转换成Python吗?我确信在Python中必须有一个单行程序来实现这一点


Tags: 代码程序算法sizestep数组javaint
2条回答

您可以使用内置的^{}方法:

size = 14
step = 4
print(list(range(0, size, step)))

输出:

[0, 4, 8, 12]

您可以使用列表理解来生成值列表

>>> groupBaseIndex  = [i for i in range(0, 13, 4)]
>>> groupBaseIndex
[0, 4, 8, 12]
>>> 

例如,您可以使用list(range())

>>> groupBaseIndex = list(range(0, 13, 4))
>>> groupBaseIndex
[0, 4, 8, 12]
>>> 

相关问题 更多 >