创建一个删除列表中奇数倍的函数

时间:2021-05-06 02:02:05

标签: python

其中一个修订问题涉及通过删除列表中所有既是奇数又是 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 吗?

2 个答案:

答案 0 :(得分:3)

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

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]

答案 1 :(得分:2)

我认为问题是在迭代列表的同时,您也在修改它(通过删除列表的元素)。

所以,试试这个:

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

欲了解更多信息 - Strange result when removing item from a list while iterating over it