首页 > 解决方案 > 创建一个删除列表中奇数倍数的函数

问题描述

其中一个修订问题涉及通过删除列表中的所有奇数和 x 倍数的整数来修改原始列表。

def remove_odd_multiples(numbers_list, multiple_of):
    for ele in numbers_list:
        if (ele%2) != 0 and (ele % multiple_of) == 0:
            numbers_list.remove(ele)
    
    return numbers_list

输出:

numbers_list = [1, 5, 23, 3, 6, 17, 9, 18]
print("Before:", numbers_list)
remove_odd_multiples(numbers_list, 3)
print("After:", numbers_list)

Before: [1, 5, 23, 3, 6, 17, 9, 18]
After: [1, 5, 23, 6, 17, 18]

它确实有效,但是将代码输入到 coderunner 中,我的代码未能通过一些未显示的隐藏检查。我打算错误地解决这个问题吗?我应该使用 pop 而不是 .remove 吗?

标签: python

解决方案


您可以使用列表推导返回一个带有结果的新列表,而不是在迭代时删除。

def remove_odd_multiples(numbers_list, multiple_of):
    return [x for x in numbers_list if x % 2 == 0 or x % multiple_of != 0]

要就地修改列表:

def remove_odd_multiples(numbers_list, multiple_of):
    numbers_list[:] = [x for x in numbers_list if x % 2 == 0 or x % multiple_of != 0]

在删除元素的同时向后循环还可以确保未处理的元素不会移动。

def remove_odd_multiples(numbers_list, multiple_of):
    for i in range(len(numbers_list) - 1, -1, -1):
        if (numbers_list[i]%2) != 0 and (numbers_list[i] % multiple_of) == 0:
            del numbers_list[i]

推荐阅读