python列表比较和替换

时间:2013-06-27 02:11:17

标签: python list

我有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])

似乎不起作用。任何人都可以帮我这个吗?

3 个答案:

答案 0 :(得分:7)

您应该使用list comprehensions

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

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

x if cond else y是Python等同于ternary operator

[ f(x) for x in xs ]生成一个新列表,其中函数f已应用于集合xs中的每个元素。

答案 1 :(得分: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]]

答案 2 :(得分:0)

@DaoWen的解决方案很不错,为了提高效率,您可以将列表元素预先转换为set s:

>>> 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]]