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

2019年8月3日 138次阅读

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

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

例子：
pattern =“abba”,str =“dog cat cat dog”应该返回true;狗是一只猫,猫是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()返回一个迭代器对象,而不是一个列表.这些对象之间的等式测试将不起作用(相等性测试身份,而不是内存中的完全相同的对象).

明确转换为列表：

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(),我们确保我们可以检测到模式长度和字数不匹配的情况.

点赞