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

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

举个例子，现有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))

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

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

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))

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

可以这样，你单纯用迭代索引a是不行的，可以使用下标

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

a = [1,2,3,4,5,6,7,8,9]
print(kmom(a,3))

你好根据你的代码，我这里提供两种解决方案：

- 1. 添加 if 过滤剩余

- 2. 主动删除多余的部分

 

def kmom(a, k):
    lst = []
    for i in a:
        if(i+k <= len(a)):  # 如果剩余长度不足，停止添加
            lst.append(a[i:i+k])
    return lst
 
def kmom2(a, k):
    lst = []
    for i in a:
        lst.append(a[i:i+k])
    lst = lst[0:-k+1]   # 删掉末尾多出的 k-1 个元素
    return lst

 

您好，我是有问必答小助手，您的问题已经有小伙伴帮您解答，感谢您对有问必答的支持与关注！
PS：问答VIP年卡 【限时加赠：IT技术图书免费领】，了解详情>>> https://vip.csdn.net/askvip?utm_source=1146287632