Python2 和 Python3 使用带有查找和索引的映射时的区别

给定一个模式和一个字符串str, 求如果str遵循相同的模式。

这里的 follow 表示完全匹配，这样在pattern中的字母和pattern中的非空单词之间存在双射str.

例子：pattern = "abba", str = "dog cat cat dog"应该返回 true；dog is a, cat is b这些词构成了abba图案。pattern = "abba", str = "dog cat cat fish"应该返回 false；该字符串后面跟着一个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当尝试测试上述示例时。有人可以给一些建议吗？

在Python 3中，map()返回一个迭代器对象，而不是一个列表。这些对象之间的相等测试将不起作用（相等是测试identity，内存中完全相同的对象）。

显式转换为列表：

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()功能 https://docs.python.org/3/library/functions.html#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() https://docs.python.org/3/library/itertools.html#itertools.starmap测试是否与operator.eq()功能 https://docs.python.org/3/library/operator.html#operator.eq。通过使用itertools.zip_longest() https://docs.python.org/3/library/itertools.html#itertools.zip_longest我们确保能够检测到模式长度和字数不匹配的情况。