比较 python 中的旋转列表

我试图比较两个列表以确定一个列表是否是另一个列表的旋转（循环排列），例如：

a = [1, 2, 3]
b = [1, 2, 3] or [2, 3, 1] or [3, 1, 2]

都是匹配的，而：

b = [3, 2, 1] is not

为此，我有以下代码：

def _matching_lists(a, b):
    return not [i for i, j in zip(a,b) if i != j]

def _compare_rotated_lists(a, b):
    rotations = [b[i:] + b[:i] for i in range(len(b))]
    matches = [i for i in range(len(rotations)) if _matching_lists(a, rotations[i])]
    return matches

这会构建 b 的所有可能旋转的列表，然后比较每个旋转。是否可以在不构建中间列表的情况下做到这一点？性能并不重要，因为列表通常只有四个项目长。我最关心的是代码的清晰度。

列表的长度始终相同。

最佳答案（保留匹配轮换列表）似乎是：

def _compare_rotated_lists(a, b):
    return [i for i in range(len(b)) if a == b[i:] + b[:i]]

你不需要这个功能_matching_lists，因为你可以使用==:

>>> [1,2,3] == [1,2,3]
True
>>> [1,2,3] == [3,1,2]
False

我建议使用any() http://docs.python.org/3/library/functions.html#any找到匹配项后立即返回，并使用生成器表达式来避免在内存中构造旋转列表：

def _compare_rotated_lists(a, b):
    """Return `True` if the list `a` is equal to a rotation of the list `b`."""
    return any(a == b[i:] + b[:i] for i in range(len(b)))

您可能会考虑检查列表的长度是否相同，以快速拒绝简单的情况。

    return len(a) == len(b) and any(a == b[i:] + b[:i] for i in range(len(b)))

正如评论中所讨论的，如果您知道a and b是可散列的，您可以使用进行初始比较collections.Counter http://docs.python.org/3/library/collections.html#collections.Counter:

    return Counter(a) == Counter(b) and any(a == b[i:] + b[:i] for i in range(len(b)))

如果你知道a and b具有可比性，您可以使用以下方法进行初步比较sorted:

    return sorted(a) == sorted(b) and any(a == b[i:] + b[:i] for i in range(len(b)))