Я хочу сделать следующее:
У меня есть два списка 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 ответа
Один из подходов состоит в том, чтобы использовать выражение генератора для перебора обоих списков, начиная с конца, и сохранить первый индекс, где найдено совпадение:
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]
Вы можете перебрать копию списка, которая обратится к оригиналу, а затем перебрать копию и удалить элементы из оригинала, как эта функция:
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
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
Похожие вопросы
Новые вопросы
python
Python - это многопарадигмальный, динамически типизированный, многоцелевой язык программирования. Он разработан для быстрого изучения, понимания и использования, а также для обеспечения чистого и единообразного синтаксиса. Обратите внимание, что Python 2 официально не поддерживается с 01.01.2020. Тем не менее, для вопросов о Python, связанных с версией, добавьте тег [python-2.7] или [python-3.x]. При использовании варианта Python (например, Jython, PyPy) или библиотеки (например, Pandas и NumPy) включите его в теги.
En Español