Python列表操作

Question

我有两个列表 ListA 和 ListB 如下：

ListA=['1','1','2','2','2','3','4','4','5','5','5','5']
ListB=['1','5']

我正在尝试提出与 List A 长度相同的 List C，但将 List A 中的数字替换为'X' 如果数字在 List B 中。我期待的结果：

ListC=['X','X','2','2','2','3','4','4','X','X','X','X']

仅供参考，ListB 的长度将始终小于 ListA 的长度，并且 ListB 将不包含任何不在 列表 A.

我试过这样：

ListA=['1','1','2','2','2','3','4','4','5','5','5','5']
ListB=['1','5']
ListC=[]
for items in ListB:
    for a in ListA:
        if items==a:
            ListC.append('X')
        else:
            ListC.append(a)

显然这将创建一个具有 (listB X 长度 A) 的列表，而不仅仅是列表 A 的长度。是否有任何内置函数可以执行此操作？谁能告诉我怎么做？

Answer 1

任何时候您在同一个list 上反复进行成员资格测试时，最好创建一个set。虽然构建 set 需要一些时间，但单独的查找速度会快得多

SetB = set(ListB)
[i if i not in SetB else 'X' for i in ListA]

Answer 2

List Comprehension 是你的朋友：

ListA=['1','1','2','2','2','3','4','4','5','5','5','5']
ListB=['1','5']
ListC = [i if i not in ListB else 'x' for i in ListA]

--> ['x', 'x', '2', '2', '2', '3', '4', '4', 'x', 'x', 'x', 'x']

希望这会有所帮助！

Answer 3

您可以使用列表推导：

[i if i not in ListB else 'X' for i in ListA]

要修复您当前的代码，请使用in 检查项目是否在ListB 中：

for item in ListA:
    if item in ListB:
        ListC.append('X')
    else:
        ListC.append(item)