python - 将 map 与查找和索引一起使用时 Python2 和 Python3 之间的区别

Question

给定一个模式和一个字符串 str，判断 str 是否遵循相同的模式。

Here follows表示完全匹配，即pattern中的字母与str中的非空词之间存在双射。

例子:pattern = "abba"，str = "dog cat cat dog" 应该返回 true； dog 是a，cat 是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。谁能给点建议？

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() function 进行比较来完全避免创建列表:

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() function 是否相等.通过使用 itertools.zip_longest()我们确保能够检测到模式长度和字数不匹配的情况。