Python实现将列表拆分为大小为N的块
作者：python收藏家
方法1：使用yield
yield关键字使函数能够在再次调用时返回到它停止的位置。这是与常规函数的关键区别，一个常规的函数不能回到它停止的地方。yield关键字帮助函数记住其状态，yield使函数能够挂起和恢复，同时它在挂起执行时返回一个值。
my_list = ['geeks', 'for', 'geeks', 'like',
'geeky','nerdy', 'geek', 'love',
'questions','words', 'life']
# Yield successive n-sized
# chunks from l.
def divide_chunks(l, n):
# looping till length l
for i in range(0, len(l), n):
yield l[i:i + n]
# How many elements each
# list should have
n = 5
x = list(divide_chunks(my_list, n))
print (x)输出
[['geeks', 'for', 'geeks', 'like', 'geeky'],
['nerdy', 'geek', 'love', 'questions', 'words'],
['life']]
方法2：使用for循环
在这个例子中，我们使用了Python中的循环和列表切片，这将帮助我们将列表分成块。
my_list = [1, 2, 3, 4, 5,
6, 7, 8, 9]
start = 0
end = len(my_list)
step = 3
for i in range(start, end, step):
x = i
print(my_list[x:x+step])输出
[1, 2, 3]
[4, 5, 6]
[7, 8, 9]
方法3： 使用列表解析
在Python中，将列表拆分为一行代码，将列表拆分为多个列表是一种优雅的方式。
my_list = [1, 2, 3, 4, 5,
6, 7, 8, 9]
# How many elements each
# list should have
n = 4
# using list comprehension
final = [my_list[i * n:(i + 1) * n] for i in range((len(my_list) + n - 1) // n )]
print (final)输出
[[1, 2, 3, 4], [5, 6, 7, 8], [9]]
另一种实现方式：
l = [1, 2, 3, 4, 5, 6, 7, 8, 9] # How many elements each # list should have n = 4 # using list comprehension x = [l[i:i + n] for i in range(0, len(l), n)] print(x)
输出
[[1, 2, 3, 4], [5, 6, 7, 8], [9]]
方法4：使用Numpy
在这里，我们使用Numpy.array_split，它将数组拆分为n个大小相等的块。
import numpy as np arr = range(30) np.array_split(arr, 6)
输出
[array([0, 1, 2, 3, 4]), array([5, 6, 7, 8, 9]), array([10, 11, 12, 13, 14]), array([15, 16, 17, 18, 19]), array([20, 21, 22, 23, 24]), array([25, 26, 27, 28, 29])]
方法5：使用itertools
from itertools import islice
def chunk(arr_range, arr_size):
arr_range = iter(arr_range)
return iter(lambda: tuple(islice(arr_range, arr_size)), ())
print(list(chunk(range(30), 5)))输出
[(0, 1, 2, 3, 4),
(5, 6, 7, 8, 9),
(10, 11, 12, 13, 14),
(15, 16, 17, 18, 19),
(20, 21, 22, 23, 24),
(25, 26, 27, 28, 29)]
方法6： 使用collections
from collections import deque
def split_list(input_list, chunk_size):
# Create a deque object from the input list
deque_obj = deque(input_list)
# While the deque object is not empty
while deque_obj:
# Pop chunk_size elements from the left side of the deque object
# and append them to the chunk list
chunk = []
for _ in range(chunk_size):
if deque_obj:
chunk.append(deque_obj.popleft())
# Yield the chunk
yield chunk
input_list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
chunk_size = 3
chunks = list(split_list(input_list, chunk_size))
print(chunks) 输出
[[1, 2, 3], [4, 5, 6], [7, 8, 9], [10]]
deque类允许您轻松地从列表的左侧或右侧移除元素，从而轻松地将列表分割为特定大小的块。代码使用while循环和生成器函数迭代列表，每次生成一个块。当deque为空时，循环中断，这表明所有元素都已被处理。
方法7： 部分赋值
这里有一个例子，你可以轻松地处理大小为N的块列表：
my_list = list(range(10))
chunk_size = 3
while my_list:
chunk, my_list = my_list[:chunk_size], my_list[chunk_size:]
print(chunk)输出
[0, 1, 2]
[3, 4, 5]
[6, 7, 8]
[9]
到此这篇关于Python实现将列表拆分为大小为N的块的文章就介绍到这了,更多相关Python拆分列表内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！