我要完成以下任务:
我有两个列表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进行编程,但是如果我在其他地方使用它,则可能会为zip创建变量,而不是上面的这一行很难理解。尽管如此,对于此答案,我宁愿尽可能短的答案而不是可读性,尽管这不是此问题的要求。只是想知道如何完成它。
答案 0 :(得分:3)
一种方法是使用生成器表达式从头开始遍历两个列表,并保留找到匹配项的第一个索引:
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]
答案 1 :(得分:2)
a=[2,3,2,2,1]
b=[2,3,4,1,1]
解决方案1:使用while循环
注意:异常处理(try-except块),以避免:IndexError:列表索引超出范围,在特殊情况下,例如,如果您有a = [1,2,3,4,5], b = [1,2,3,4,5]
try:
while a[-1] == b[-1]:
a.pop()
b.pop()
except:
pass
print (a)
print (b)
或
while a and a[-1] == b[-1]:
a.pop()
b.pop()
print (a)
print (b)
结果:
in: a=[2,3,2,2,1], b=[2,3,4,1,1]
out: [2, 3, 2, 2],[2, 3, 4, 1]
in: a=[10,10,10,10,10],b=[10,10,10,10,9]
out: [10, 10, 10, 10, 10],[10, 10, 10, 10, 9]
in: a=[9,10,10,10,10],b=[10,10,10,10,10]
out: [9],[10]
in: a=[1,2,3,4,5],b=[1,2,3,4,5]
out: [], []
解决方案2:使用递归
def remove(a,b):
if a[-1] == b[-1]:
a.pop()
b.pop()
return remove(a,b)
# else:
# return
remove(a,b)
print (a)
print (b)
Python slice()
slice()构造函数创建一个切片对象,该对象表示一组 由范围(开始,停止,步进)指定的索引。
a[-1] # return a last element of list
Python列表pop()
pop()方法从列表中删除给定索引处的项目。 该方法还返回已删除的项目。
pop()方法的语法是:
list.pop(index)
a.pop() # removing last element of list
答案 2 :(得分:0)
您可以遍历与原始文档相反的列表的副本,然后遍历该副本并从原始文档中删除元素,如下所示:
class SomeClass:
def removeSameCharacters(a, b):
x = a.reverse
y = b.reverse
for i in x:
if x[i] == y[i]:
a.remove[i]
b.remove[i]
else:
break