gpt4 book ai didi

python - 创建每 n 个元素跳过 k 个元素的整数序列

转载 作者:行者123 更新时间:2023-12-01 05:02:45 25 4
gpt4 key购买 nike

在 numpy 中,创建整数序列以便省略每个 n 个元素 k 个数字的最简单/最优雅的方法是什么?

序列很大,我想尽可能避免 for 循环。

例如:

n = 4
k = 2

desired output = {0,1,2,3,6,7,8,9,12,13,14,15,...}

最佳答案

seq = numpy.arange((n+k)*10).reshape(-1,n+k)[:,:n].flatten()

关于python - 创建每 n 个元素跳过 k 个元素的整数序列,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/25709561/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com