Python 如果列表的最后 N 个值与列表 2 的最后 N 个值相同

Question

我正在练习过去的纸质问题，以便复习下周的考试

假设我有两个列表

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

并将变量 n 设置为一个值，即 2我需要创建一个函数来检查 list1 和 list2 中最后的“n”值是否以相同的顺序相同。

一种使用 for 循环的变体和一种不使用循环的变体

在这种情况下，它将返回 true，因为列表 1 和 2 中的 3,4 相同

如果 b=[1,2,4,3] 则返回 false

我当前没有循环的代码是:

def compare1(a,b,n):
    if a[-n:]==b[-n:]:
        return True
    else:
        return False

我相信当我测试它时它会起作用

但是我如何将其转换为 for 循环呢？如果我添加:

def compare2(a,b,n):
    for i in a and b:
        if a[-n:]==b[-n:]:
            return True
        else:
            return False

这可以工作，但 for 循环没有做任何事情

Answer 1

有几种方法可以做到这一点。您可以循环遍历列表的索引，也可以将列表压缩在一起并直接循环遍历项目。这里有一些例子。我还优化了您的 compare1 函数。

def compare1(a, b, n):
    return a[-n:] == b[-n:]

def compare2(a, b, n):
    for i in range(-n, 0, 1):
        if a[i] != b[i]:
            return False
    return True

def compare3(a, b, n):
    for u, v in zip(a[-n:], b[-n:]):
        if u != v:
            return False
    return True

def compare4(a, b, n):
    return all(u == v for u, v in zip(a[-n:], b[-n:]))

# test

funcs = (compare1, compare2, compare3, compare4)

a = [1, 2, 3, 4]
b = [5, 6, 3, 4]
c = [1, 2, 5, 6]

for compare in funcs:
    print(compare.__name__)
    print(a, b, compare(a, b, 2))
    print(a, c, compare(a, c, 2))
    print()

输出

compare1
[1, 2, 3, 4] [5, 6, 3, 4] True
[1, 2, 3, 4] [1, 2, 5, 6] False

compare2
[1, 2, 3, 4] [5, 6, 3, 4] True
[1, 2, 3, 4] [1, 2, 5, 6] False

compare3
[1, 2, 3, 4] [5, 6, 3, 4] True
[1, 2, 3, 4] [1, 2, 5, 6] False

compare4
[1, 2, 3, 4] [5, 6, 3, 4] True
[1, 2, 3, 4] [1, 2, 5, 6] False

FWIW，compare1 是执行此操作的最佳方法，因为它在 C 级别执行所有循环和测试，而不是使用显式 Python 代码。