使用带有find和index的map时Python2和Python3之间的区别

Question

给定一个模式和一个字符串str,找到是否str遵循相同的模式.

以下是完全匹配,这样在模式中的字母和非空单词之间存在双射str.

示例: pattern = "abba",str = "dog cat cat dog"应该返回true; dog是a,cat是b和形成abba模式的词. pattern = "abba",str = "dog cat cat fish"应该返回虚假; 字符串遵循abbc模式.

我的解决方案适用于Python 2:

def wordPattern(self, pattern, str):
    s = pattern
    t = str.split()
    return map(s.find, s) == map(t.index, t)

但我只是想知道为什么这个解决方案在Python 3中不起作用.在那里,False当试图测试上面的例子时,该函数总会返回.有人可以请一些建议吗？

Answer 1

在Python 3中,map()返回迭代器对象,而不是列表.这些对象之间的等式测试将不起作用(相等性测试身份,而不是内存中的完全相同的对象).

明确转换为列表:

def wordPattern(self, pattern, str):
    s = pattern
    t = str.split()
    return list(map(s.find, s)) == list(map(t.index, t))

或使用列表推导:

def wordPattern(self, pattern, str):
    s = pattern
    t = str.split()
    return [s.find(c) for c in  s] == [t.index(w) for w in t]

或者通过将压缩结果与all()函数进行比较来完全避免创建列表:

from operator import eq
from itertools import starmap, zip_longest

def wordPattern(self, pattern, str):
    s = pattern
    t = str.split()
    return all(starmap(eq, zip_longest(map(s.find, s), map(t.index, t))))

如果没有匹配,后者短路而不必进行所有比较.本着保持功能风格的精神,我曾经itertools.starmap()测试过与operator.eq()函数的相等性.通过使用itertools.zip_longest()我们确保我们可以检测到模式长度和字数不匹配的情况.