如何将python列表拆分为大小相等的块
问：
如何将任意长度的列表拆分为大小相等的块？
答：
简单的实现，对于 Python3：
def chunks(lst, n):
n = max(1, n)
return [lst[i:i+n] for i in range(0, len(lst), n)]测试截图如下：
对于 Python2，改用如下代码：
def chunks(lst, n):
n = max(1, n)
return [lst[i:i+n] for i in xrange(0, len(lst), n)]
还可使用 itertools 模块的 islice 方法来实现所需功能:
from itertools import islice
def batched(iterable, n):
"Batch data into lists of length n. The last batch may be shorter."
# batched('ABCDEFG', 3) --> ABC DEF G
it = iter(iterable)
while True:
batch = list(islice(it, n))
if not batch:
return
yield batch将数据分批放到到长度为 n 的列表中，最后一批可能更短。
测试截图如下：
另外，可以使用第三方库 numpy 中的拆分数组方法 array_split：
将数组拆分为多个子数组
import numpy as np
lst = range(33)
np.array_split(lst, 8)参考：
- stackoverflow question 312443
- https://docs.python.org/3/library/itertools.html
- https://numpy.org/doc/stable/reference/generated/numpy.array_split.html
相关阅读：
本文参与 腾讯云自媒体同步曝光计划，分享自微信公众号。
原始发表：2023-08-13 21:30，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
腾讯云开发者
