python列表比较和替换

Question

我有2个清单。第listA一个列表是列表列表。listA=[[1,2,5,3],[3,1,5],[7,9,2]]第二个列表，listB是我要与listA中的其他列表进行比较的列表listB=[1,2,3,4,5,6,7,8,9,10]我想分别比较listA中的列表，如果listB中存在列表项，则用'T'代替。如果不是，请保留listB项。应该是这样的listC=[['T','T','T',4,'T',6,7,8,9,10],['T',2,'T',4,'T',6,7,8,9,10],[1,'T',3,4,5,6,'T',8,'T',10]]我已经尝试过这样的事情：for item in listA:    for i in range(10):        listC.append([i if i not in item else 'T' for i in listB])似乎不起作用。谁能帮我这个？

Answer 1

您应该使用列表推导：

listC = [ [ ('T' if b in a else b) for b in listB ] for a in listA ]

括号不是必需的，但是它们可能会使它更具可读性。

x if cond else y是Python的等价三元运算符。

[ f(x) for x in xs ]产生一个新列表，其中功能f已应用于集合中的每个元素xs。

Answer 2

尼斯和可读:)

listC = []

for i in listA:

    temp = []

    for x in listB:

        if x in i:

            temp.append('T')

        else:

            temp.append(x)

    listC.append(temp)

print listC

印刷：

[['T', 'T', 'T', 4, 'T', 6, 7, 8, 9, 10], ['T', 2, 'T', 4, 'T', 6, 7, 8, 9, 10], [1, 'T', 3, 4, 5, 6, 'T', 8, 'T', 10]]

Answer 3

为了提高效率，您可以将列表元素预先转换为sets：

>>> listA=[[1,2,5,3],[3,1,5],[7,9,2]]

>>> listB=[1,2,3,4,5,6,7,8,9,10]

>>> setA = [set(A) for A in listA]

>>> [['T' if x in A else x for x in listB] for A in setA]

[['T', 'T', 'T', 4, 'T', 6, 7, 8, 9, 10], ['T', 2, 'T', 4, 'T', 6, 7, 8, 9, 10], [1, 'T', 3, 4, 5, 6, 'T', 8, 'T', 10]]