在列表中查找匹配项和不匹配项

Question

我对 Python 还是很陌生，对列表可以做什么和不能做什么有点困惑。我有两个列表要比较并以二进制格式返回匹配和不匹配的元素。 List1 的长度不变，而List2 的长度不同（但总是比List1 短）。

例如：

List1 = ['dog', 'cat', 'pig', 'donkey']
List2 = ['dog', 'cat', 'donkey']

想要的输出：

List3 = [1, 1, 0, 1]

我目前的代码是：

def match_nonmatch(List1, List2):
    List3 = []
    for i in range(len(List1)):
        for j in range(len(List2)):
            if List1[i] == List2[j]:
                List3.append(1)
            else:
                List3.append(0)
   return List3

当我比较列表时，我能够返回匹配项，但是当我包含上面显示的 else 语句以返回不匹配项时，我最终得到的列表比它应该的要长得多。例如，当我使用一个比较 60 个项目的列表时，我得到一个包含 3600 个项目而不是 60 个项目的列表。

如果有人能向我解释我的代码目前存在的问题并建议我如何修改代码以使其达到我想要的效果，我将不胜感激。

Answer 1

你也可以使用按位运算：

List1 = ['dog', 'cat', 'pig', 'donkey']  
List2 = ['dog', 'cat', 'donkey']  

匹配：

set(List1) & set(List2)   

不匹配：

set(List1) ^ set(List2) 

Answer 2

使用set 而不是list。这样你可以做很多好事：

set1 = set(['dog', 'cat', 'pig', 'donkey'])
set2 = set(['dog', 'cat', 'donkey'])

matched = set1.intersection(set2) # set(['dog', 'cat', 'donkey'])
unmatched = set1.symmetric_difference(set2) # set(['pig'])

我知道这并不完全符合您的要求，但在执行此类操作时，使用集合而不是列表通常是一种更好的做法。

更多关于套装的信息：http://docs.python.org/library/stdtypes.html#set

Answer 3

使用以下代码。

listt3=[]

for i in listt1:
    if i in listt2:
        listt3.append(1)
    else:
        listt3.append(0)

如果你更喜欢单行，

listt3=[ 1 if i in listt2 else 0 for i in listt1]

Answer 4

如果list2 很短，我会这样做：

list1 = ['dog', 'cat', 'pig', 'donkey']
list2 = ['dog', 'cat', 'donkey']
list3 = [int(val in list2) for val in list1]
print(list3)

打印出来：

[1, 1, 0, 1]

如果list2 很长，可以先将其转换为set，以提高代码效率：

list1 = ['dog', 'cat', 'pig', 'donkey']
set2 = set(['dog', 'cat', 'donkey'])
list3 = [int(val in set2) for val in list1]
print(list3)

您当前代码产生过多元素的原因是您在内循环的每次迭代中都调用了append()，并且有len(List1) * len(List2) 这样的迭代。

解决方法如下：

def match_nonmatch(List1, List2):
    List3 = []
    for i in range(len(List1)):
        for j in range(len(List2)):
            if List1[i] == List2[j]:
                List3.append(1)
                break             # fix #1
        else:                     # fix #2
            List3.append(0)
    return List3

注意添加的break 以及else 子句现在指的是内部for 而不是if。

也就是说，我仍然会在答案的顶部使用单行。

Answer 5

>>> list1 = ['dog', 'cat', 'pig', 'donkey']; list2 = ['dog', 'cat', 'donkey']
>>> [i in list2 for i in list1]
[True, True, False, True]

另外，您应该阅读 PEP8，CamelCase 名称通常仅用于类。

Answer 6

[int(i==j) for i, j in zip(list1, list2)]