定义函数从列表中提取元素出现错误

Question

问题遇到的现象和发生背景

举个例子，现有a = [1,2,3,4,5,6,7,8,9]

创造函数kmom(a,k)，使得函数返回
[[1,2,3], [2,3,4], [3,4,5], ... , [7,8,9]]

问题相关代码

我尝试了两种方法定义kmom(a,k)，但都没有成功

#第一种方式
a = []
for i in range(20):
    a.append(random.randint(1, 10))
random.shuffle(a)
print(a)

def kmom(a, k):
    lst = []
    for i in a:
        lst.append(a[i:i+k])
    return lst

print(kmom(a,3))

结果并不是按照迭代顺序提取元素

#第二种方法
a = []
for i in range(20):
    a.append(random.randint(1, 10))
random.shuffle(a)
print(a)

def kmom(a, k):
    lst = []
    for i in a:
        lst.append(a[:k])
        del(a[0])
    return lst

print(kmom(a,3))

思路感觉是对的，但是只迭代了一半的列表

请问各位，图中的两种定义方式哪里出现了问题？😣
还有就是，我应当如何定义函数以达成题目

Answer 1

for i in a: 这样i遍历的是列表中的值, 应该是遍历列表的下标才对

改成
for i in range(len(a)-k+1):

你题目的解答代码如下：

import random
a = []
for i in range(20):
    a.append(random.randint(1, 10))
random.shuffle(a)
print(a)

def kmom(a, k):
    lst = []
    for i in range(len(a)-k+1):
        lst.append(a[i:i+k])
    return lst

print(kmom(a,3))

如有帮助，请点击我的回答下方的【采纳该答案】按钮帮忙采纳下，谢谢!