📜  Python – 连续K个元素加入List(1)

📅  最后修改于: 2023-12-03 15:34:09.364000             🧑  作者: Mango

Python – 连续K个元素加入List

当我们需要在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

代码解释:

  1. 定义一个add_k_elements()函数,该函数接收两个参数:列表lst和整数k
  2. 定义一个空列表res用于存储结果;
  3. 使用for循环遍历lst,每次遍历从索引i开始到索引i+k结束,将这段连续的元素加入res列表中;
  4. 返回结果列表res
方法二

我们可以使用列表生成式和range()函数来达到相同的效果。

def add_k_elements(lst, k):
    return [lst[i:i+k] for i in range(0, len(lst), k)]

代码解释:

  1. 定义一个add_k_elements()函数,该函数接收两个参数:列表lst和整数k
  2. 使用列表生成式和range()函数遍历lst,每次遍历从索引i开始到索引i+k结束,将这段连续的元素加入新列表中;
  3. 返回新列表。

两种方法都非常简单易懂,可以根据自己的需求来选择使用哪一种。