我想完成以下任务:
我有两个列表 a 和 b 保证大小为 5。我现在想从两个列表的末尾删除值相同的索引/压缩/转置时。作为输入和预期输出的示例:
In: a=[2,3,2,2,1], b=[2,3,4,1,1]
Out: a=[2,3,2,2], b=[2,3,4,1]
In: a=[9,10,10,10,10], b=[10,10,10,10,10]
Out: a=[9], b=[10]
In: a=[1,2,3,4,5], b=[1,2,3,4,5]
Out: a=[], b=[]
# (a=[1], b=[1] or a=[1,2,3,4,5], b[1,2,3,4,5] are fine as well
# for this last example, as long as there isn't any error)
In: a=[10,10,10,10,10], b=[10,10,10,10,9]
Out: a=[10,10,10,10,10], b=[10,10,10,10,9]
我知道如何删除在相同索引处相等的所有值:
f = lambda a,b: [] if a==b else map(list, zip(*[(i,j) for(i,j) in zip(a,b) if i!=j]))[0]
然后我可以将其称为:
a,b = [2,3,2,2,1], [2,3,4,1,1]
A,B = f(a,b), f(b,a)
但这会导致 A=[2,2], B=[4,1]，同时删除前导值。
在同一索引处发现不匹配之前，从两个列表中删除尾随值的最简单方法是什么？
PS:这是一个 code-golf挑战。我几乎从不使用 Python 编程，但如果我要在其他地方使用它，我可能会为 zips 创建变量，而不是我上面的这行非常难读的单行。不过，对于这个答案，我更喜欢尽可能简短的答案而不是可读性，尽管这不是这个问题的要求。只是想知道一般如何完成它。
请您参考如下方法:
一种方法是使用生成器表达式从末尾开始迭代两个列表，并保留找到匹配项的第一个索引:
a=[2,3,2,2,1]
b=[2,3,4,1,1]
ix = next((ix for ix,(i,j) in enumerate(zip(a[::-1],b[::-1])) if i != j), None)
然后您可以使用它来对列表进行切片(使用 if 语句检查返回值是否为 None，这意味着两个列表相等):
if ix:
print(a[:len(a)-ix])
print(b[:len(b)-ix])
# [2, 3, 2, 2]
# [2, 3, 4, 1]
对于你的另一个例子:
a=[9,10,10,10,10]
b=[10,10,10,10,10]
ix = next(ix for ix,(i,j) in enumerate(zip(a[::-1],b[::-1])) if i != j)
if ix:
print(a[:len(a)-ix])
print(b[:len(b)-ix])
# [9]
# [10]
