python - 接收列表 (myList) 和值 (n) 并检查该值是否在该列表中的函数

Question

我想知道是否有新人如何用 python 做这个问题？编写一个函数，它接受列表 (myList) 和值 (n)，并检查该值是否在该列表中。如果该值确实存在于列表中，则将其替换为值“1”。您的函数应该返回更新后的列表。使用函数签名:def ReplaceNum(myList, n):

示例:如果函数的输入为replaceNum([12, 6, 10, 10], 10)，则函数将返回包含[12, 6, 1, 1]的列表。

到目前为止我已经有了

def replaceNum(myList,n):
    new_list = []
    for i in range(0, len(myList)):
        if i == n:
            myList[n] = 1
            new_list.append(i)

        return myList[i]
print(replaceNum([5, 10, 20], 10))

但它只打印 5，我需要它打印 [5,1,20]

Answer 1

您可以使用list comprehension来做到这一点。该函数迭代 myList 中的每个元素，如果它等于 n，则将其替换为 1:

def replaceNum(myList, n):
    return [x if x != n else 1 for x in myList] 

print(replaceNum([5, 10, 20], 10))

输出

[5, 1, 20]