📅  最后修改于: 2023-12-03 15:34:09.364000             🧑  作者: Mango
当我们需要在Python中给定一个列表,并且需要将连续的K个元素加入一个新的列表时,我们可以使用列表切片的方法来实现。
我们可以使用for循环和列表切片来遍历列表,并将连续的K个元素加入新的列表中。
def add_k_elements(lst, k):
res = []
for i in range(0, len(lst), k):
res.append(lst[i:i+k])
return res
代码解释:
add_k_elements()
函数,该函数接收两个参数:列表lst
和整数k
;res
用于存储结果;lst
,每次遍历从索引i
开始到索引i+k
结束,将这段连续的元素加入res
列表中;res
。我们可以使用列表生成式和range()
函数来达到相同的效果。
def add_k_elements(lst, k):
return [lst[i:i+k] for i in range(0, len(lst), k)]
代码解释:
add_k_elements()
函数,该函数接收两个参数:列表lst
和整数k
;range()
函数遍历lst
,每次遍历从索引i
开始到索引i+k
结束,将这段连续的元素加入新列表中;两种方法都非常简单易懂,可以根据自己的需求来选择使用哪一种。