python – 在列表中的每n个项目中获取一大块项目

发布时间：2020-12-20 11:08:51 所属栏目：Python 来源：网络整理

导读：说我有以下列表： l = [4,3,1,5,8,11,10,4,12,2,1] 切片l的最pythonic方法是什么,以获得长度为n的块,在块之间留下k个项目？例如,如果n = 2且k = 3,则结果应为： [4,2] 解决方法使用列表理解： [e for i in range(0,len(l),n+k) for e in l[i:i+n]]# [4,2]

说我有以下列表：

l = [4,3,1,5,8,11,10,4,12,2,1]

切片l的最pythonic方法是什么,以获得长度为n的块,在块之间留下k个项目？

例如,如果n = 2且k = 3,则结果应为：

[4,2]

解决方法

使用列表理解：

[e for i in range(0,len(l),n+k) for e in l[i:i+n]]
# [4,2]

一个笨拙的解决方案是：

import numpy as np

idx = (np.arange(0,n+k)[:,None] + np.arange(n)).ravel()
np.array(l)[idx]
# array([ 4,2])

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!