Python идиома для итерации изменения списка

Question

Есть ли лучше (очевидно,/идиоматический) способ питона писать эквивалент

index = 0 
while index < len(some_list): 
    do_some_stuff(some_list[index]) # may have many side effects 
    if delete_element(some_list[index]): 
     del some_list[index] 
    else: 
     index += 1 

или словарную эквивалент этого для словарей? Dict/list comprehension - это не вещь, потому что результат delete_element может зависеть от do_some_stuff.

Answer 1

Если заказ не имеет значения, вы можете перечислить его в обратном порядке, чтобы удалять не испортили часть списка, который вы еще не обработали.

for i, item in enumerate(reversed(somelist), -len(somelist)+1): 
    do_some_stuff(item) 
    if delete_element(item): 
     del somelist[-i] 

Если заказ имеет значение, отмените список, сделайте этот трюк, а затем снова отмените его. Это смутит их!

В зависимости от ситуации вы можете заменить элемент маркером, таким как None. Либо выпустите его позже, либо используйте другие места, в котором этот список используется.

for i, item in enumerate(somelist)): 
    do_some_stuff(item) 
    if delete_element(item): 
     somelist[i] = None 

somelist = [item for item in somelist if item] 

Answer 2

Вы можете разбить операции на две отдельные петли и использовать представление списка для второй части.

for value in some_list: 
    do_some_stuff(value) 

some_list = [value for value in some_list if not delete_element(value)] 

Другим решением было бы перебрать копию списка и использовать enumerate для отслеживания индексов без необходимости поддерживать счетчик вручную.

for index, value in enumerate(some_list[::-1]): 
    do_some_stuff(value) 
    if delete_element(value): 
     del some_list[-index - 1] 

Вы хотели бы итерацию назад, так что вам не придется настраивать index для удаленных элементов.

Answer 3

вы могли бы даже, если предположить, что do_some_stuff возвращается item, сделать что-то вроде:

res = [item for item in some_list if not delete_element(do_some_stuff(item))] 

Answer 4

без излишеств способ заключается в использовании генератор:

def do_stuff_and_filter_items(items): 
    for item in items: 
     do_some_stuff(item) 
     if not delete_element(item): 
      yield item 

Тогда, чтобы получить истребитель:

items = do_stuff_and_filter_items(items) 

или получить список:

items = list(do_stuff_and_filter_items(items)) 

или перезаписать старый список:

items[:] = do_stuff_and_filter_items(items)