Удалить конечные элементы двух списков, если значения по одному и тому же индексу равны

Я хочу выполнить следующее:

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


python transpose equality python-2.x trailing
person Kevin Cruijssen    schedule 30.04.2019    source источник

Ответы (3)


arrow_upward
3
arrow_downward

Один из подходов заключается в использовании выражения генератора для перебора обоих списков, начиная с конца, и сохранения первого индекса, в котором найдено совпадение:

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]
person yatu    schedule 30.04.2019
comment
А, это выглядит довольно хорошо. Кстати, разве i-j != 0 не просто i != j? - person Kevin Cruijssen; 30.04.2019
comment
Хм, похоже, это не работает для a=[10,10,10,10,10], b=[10,10,10,10,9] (что означает, что a и b остаются прежними, но в настоящее время они приводят к [] после a[:-ix] и b[:ix]). Забыл об этом тестовом примере. Добавлю его к вопросу. Что было бы простым решением для этого? - person Kevin Cruijssen; 30.04.2019
comment
Да, правда, это связано с окончательной нарезкой. Добавлено быстрое исправление @KevinCruijssen - person yatu; 30.04.2019
comment
также значение по умолчанию None должно быть передано в next, чтобы избежать ошибки, когда a == b - person Lante Dellarovere; 30.04.2019
comment
Собирался сделать комментарий о том, что a[len(a):-ix] не работает, но я вижу, вы отредактировали его на a[:len(a)-ix], который работает как шарм. :) Это не работает, только если a и b полностью равны, но это легко исправить с помощью [] if a==b else . Спасибо! Приму как ответ. - person Kevin Cruijssen; 30.04.2019
comment
Да, я также обновляюсь, чтобы исправить это, спасибо, что указал @LanteDellarovere - person yatu; 30.04.2019
comment
@yatu Ах, отлично. Seems to work like a charm for all my test cases (after changing None to 0 for мой вариант использования). :) Принято, и еще раз спасибо! - person Kevin Cruijssen; 30.04.2019

arrow_upward
2
arrow_downward 

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)

or 

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 List pop()

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

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

list.pop(index)

a.pop() # removing last element of list
person ncica    schedule 30.04.2019
comment
да, это необходимо для особых случаев! если у вас есть: a=[1,2,3,4,5] b=[1,2,3,4,5], без блока try-except вы получите: IndexError: индекс списка вне диапазона @Chris_Rands - person ncica; 30.04.2019
comment
я имел в виду while a and a[-1] == b[-1]: - person Chris_Rands; 30.04.2019
comment
в таком случае согласен! :) @Chris_Rands - person ncica; 30.04.2019
comment
Тоже хороший ответ! Я думаю, что a and должен быть частью if и в рекурсивном подходе, не так ли? В случае, если a и b равны. - person Kevin Cruijssen; 30.04.2019

arrow_upward
0
arrow_downward

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

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
person wulff_1996    schedule 30.04.2019