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

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

1 голос
/ 30 апреля 2019

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

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

Ответы [ 3 ]

3 голосов
/ 30 апреля 2019

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

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]
2 голосов
/ 30 апреля 2019 
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 () создает объект слайса, представляющий набор индексов, заданных диапазоном (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
0 голосов
/ 30 апреля 2019

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

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
...