Я хочу сделать следующее:

У меня есть два списка 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-ов вместо этой совершенно нечитаемой отдельной строки, которую я имел выше. Тем не менее, для этого ответа я бы предпочел как можно более короткие ответы, а не читабельность, хотя для этого вопроса это не является обязательным требованием. Просто хочу знать, как этого добиться в целом.

1
Kevin Cruijssen 30 Апр 2019 в 18:17

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]
3
yatu 30 Апр 2019 в 16:08

Вы можете перебрать копию списка, которая обратится к оригиналу, а затем перебрать копию и удалить элементы из оригинала, как эта функция:

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
0
wulff_1996 30 Апр 2019 в 16:25
a=[2,3,2,2,1]
b=[2,3,4,1,1]

решение 1: используйте цикл while

ПРИМЕЧАНИЕ. Обработка исключений (блок try-Кроме), чтобы избежать: 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 () создает объект слайса, представляющий набор индексов, заданных диапазоном (start, stop, step).

 a[-1] # return a last element of list

Список Python pop ()

Метод pop () удаляет элемент по указанному индексу из списка. Метод также возвращает удаленный элемент.

Синтаксис метода pop ():

list.pop(index)

a.pop() # removing last element of list
2
ncica 30 Апр 2019 в 15:53