Удаление нескольких элементов из списка

Можно ли удалить несколько элементов из списка одновременно? Если я захочу удалить элементы с индексами 0 и 2 и попробовать что-то вроде del somelist[0], а затем del somelist[2]второе утверждение фактически удалит somelist[3].

Я полагаю, что я всегда мог сначала удалить элементы с более высоким номером, но я надеюсь, что есть лучший способ.

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

indices = 0, 2
somelist = [i for j, i in enumerate(somelist) if j not in indices]

По какой-то причине мне не нравится ни один из ответов здесь. Да, они работают, но, строго говоря, большинство из них не удаляют элементы в списке, не так ли? (Но сделав копию, а затем заменив оригинал отредактированной копией).

Почему бы просто не удалить сначала более высокий индекс?

Для этого есть причина? Я бы просто сделал:

for i in sorted(indices, reverse=True):
    del somelist[i]

Если вы действительно не хотите удалять элементы задом наперед, тогда, я думаю, вам следует просто деинкрементировать значения индексов, которые больше, чем последний удаленный индекс (на самом деле нельзя использовать тот же индекс, поскольку у вас другой список), или использовать копия списка (которая не будет «удалять», а заменять оригинал отредактированной копией).

Я что-то здесь упускаю, есть ли причина НЕ удалять в обратном порядке?

Если вы удаляете несколько несмежных элементов, то то, что вы описываете, является наилучшим способом (и да, обязательно начните с самого высокого индекса).

Если ваши элементы находятся рядом, вы можете использовать синтаксис назначения слайса:

a[2:10] = []

Вы можете использовать numpy.deleteследующим образом:

import numpy as np
a = ['a', 'l', 3.14, 42, 'u']
I = [0, 2]
np.delete(a, I).tolist()
# Returns: ['l', '42', 'u']

Если вы не против получить numpyмассив в конце, вы можете опустить .tolist(). Вы также должны увидеть довольно значительные улучшения скорости, что делает это решение более масштабируемым. Я не тестировал его, но numpyоперации - это скомпилированный код, написанный на C или Fortran.

Как специализация ответа Грега, вы даже можете использовать расширенный синтаксис срезов. например. Если вы хотите удалить пункты 0 и 2:

>>> a= [0, 1, 2, 3, 4]
>>> del a[0:3:2]
>>> a
[1, 3, 4]

Конечно, это не распространяется на произвольный выбор, но он может определенно работать для удаления любых двух элементов.

Как функция:

def multi_delete(list_, *args):
    indexes = sorted(list(args), reverse=True)
    for index in indexes:
        del list_[index]
    return list_

Работает за время n log (n) , что должно сделать его самым быстрым и правильным решением.

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

Наша цель - удалить все гласные, которые предварительно рассчитаны как индексы 1, 4 и 7. Обратите внимание, что важно, чтобы индексы to_delete находились в порядке возрастания, иначе он не будет работать.

to_delete = [1, 4, 7]
target = list("hello world")
for offset, index in enumerate(to_delete):
  index -= offset
  del target[index]

Было бы сложнее, если бы вы хотели удалить элементы в любом порядке. ИМО, сортировка to_deleteможет быть проще, чем выяснить, когда вы должны или не должны вычитать из index.

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

some_list = [1,2,3,4,5,6,7,8,10]
rem = [0,5,7]

for i in rem:
    some_list[i] = '!' # mark for deletion

for i in range(0, some_list.count('!')):
    some_list.remove('!') # remove
print some_list

Очевидно, что из-за необходимости выбора символа «метка для удаления» это имеет свои ограничения.

Что касается производительности по мере масштабирования списка, я уверен, что мое решение не оптимально. Тем не менее, это просто, что, я надеюсь, понравится другим новичкам, и будет работать в простых случаях, когда some_listимеет широко известный формат, например, всегда числовой ...

Вот альтернатива, которая не использует enumerate () для создания кортежей (как в первоначальном ответе SilentGhost).

Это кажется более читабельным для меня. (Возможно, я бы почувствовал себя по-другому, если бы использовал привычку перечисления.) ПРЕДУПРЕЖДЕНИЕ. Я не проверял производительность двух подходов.

# Returns a new list. "lst" is not modified.
def delete_by_indices(lst, indices):
    indices_as_set = set(indices)
    return [ lst[i] for i in xrange(len(lst)) if i not in indices_as_set ]

ПРИМЕЧАНИЕ. Синтаксис Python 2.7. Для Python 3 xrange=> range.

Использование:

lst = [ 11*x for x in xrange(10) ]
somelist = delete_by_indices( lst, [0, 4, 5])

somelist:

[11, 22, 33, 66, 77, 88, 99]

--- БОНУС ---

Удалить несколько значений из списка. То есть у нас есть значения, которые мы хотим удалить:

# Returns a new list. "lst" is not modified.
def delete__by_values(lst, values):
    values_as_set = set(values)
    return [ x for x in lst if x not in values_as_set ]

Использование:

somelist = delete__by_values( lst, [0, 44, 55] )

somelist:

[11, 22, 33, 66, 77, 88, 99]

Это тот же ответ, что и раньше, но на этот раз мы предоставили ЗНАЧЕНИЯ для удаления [0, 44, 55].

Альтернативный метод понимания списка, который использует значения индекса списка:

stuff = ['a', 'b', 'c', 'd', 'e', 'f', 'woof']
index = [0, 3, 6]
new = [i for i in stuff if stuff.index(i) not in index]

Это возвращает:

['b', 'c', 'e', 'f']

Вот еще один метод, который удаляет элементы на месте. Кроме того, если ваш список очень длинный, это быстрее.

>>> a = range(10)
>>> remove = [0,4,5]
>>> from collections import deque
>>> deque((list.pop(a, i) for i in sorted(remove, reverse=True)), maxlen=0)

>>> timeit.timeit('[i for j, i in enumerate(a) if j not in remove]', setup='import random;remove=[random.randrange(100000) for i in range(100)]; a = range(100000)', number=1)
0.1704120635986328

>>> timeit.timeit('deque((list.pop(a, i) for i in sorted(remove, reverse=True)), maxlen=0)', setup='from collections import deque;import random;remove=[random.randrange(100000) for i in range(100)]; a = range(100000)', number=1)
0.004853963851928711

Это было упомянуто, но так или иначе никто не сумел действительно понять это правильно.

По O(n)решению будет:

indices = {0, 2}
somelist = [i for j, i in enumerate(somelist) if j not in indices]

Это действительно близко к версии SilentGhost , но добавляет две скобки.

l = ['a','b','a','c','a','d']
to_remove = [1, 3]
[l[i] for i in range(0, len(l)) if i not in to_remove])

По сути, это то же самое, что и ответ с наибольшим количеством голосов, просто другой способ его написания. Обратите внимание, что использование l.index () не очень хорошая идея, потому что он не может обрабатывать дублированные элементы в списке.

Метод удаления вызовет большое смещение элементов списка. Я думаю, что лучше сделать копию:

...
new_list = []
for el in obj.my_list:
   if condition_is_true(el):
      new_list.append(el)
del obj.my_list
obj.my_list = new_list
...

технически, ответ НЕТ, невозможно удалить два объекта ОДНОВРЕМЕННО. Тем не менее, возможно удалить два объекта в одной строке красивого питона.

del (foo['bar'],foo['baz'])

восстановим foo['bar'], потомfoo['baz']

мы можем сделать это с помощью цикла for, перебирающего индексы после сортировки списка индексов в порядке убывания

mylist=[66.25, 333, 1, 4, 6, 7, 8, 56, 8769, 65]
indexes = 4,6
indexes = sorted(indexes, reverse=True)
for i in index:
    mylist.pop(i)
print mylist

Для индексов 0 и 2 из списка А:

for x in (2,0): listA.pop(x)

Для удаления некоторых случайных индексов из списка A:

indices=(5,3,2,7,0) 
for x in sorted(indices)[::-1]: listA.pop(x)

Я хотел сравнить различные решения, которые позволили легко поворачивать ручки.

Сначала я сгенерировал свои данные:

import random

N = 16 * 1024
x = range(N)
random.shuffle(x)
y = random.sample(range(N), N / 10)

Затем я определил свои функции:

def list_set(value_list, index_list):
    index_list = set(index_list)
    result = [value for index, value in enumerate(value_list) if index not in index_list]
    return result

def list_del(value_list, index_list):
    for index in sorted(index_list, reverse=True):
        del(value_list[index])

def list_pop(value_list, index_list):
    for index in sorted(index_list, reverse=True):
        value_list.pop(index)

Тогда я использовал timeitдля сравнения решений:

import timeit
from collections import OrderedDict

M = 1000
setup = 'from __main__ import x, y, list_set, list_del, list_pop'
statement_dict = OrderedDict([
    ('overhead',  'a = x[:]'),
    ('set', 'a = x[:]; list_set(a, y)'),
    ('del', 'a = x[:]; list_del(a, y)'),
    ('pop', 'a = x[:]; list_pop(a, y)'),
])

overhead = None
result_dict = OrderedDict()
for name, statement in statement_dict.iteritems():
    result = timeit.timeit(statement, number=M, setup=setup)
    if overhead is None:
        overhead = result
    else:
        result = result - overhead
        result_dict[name] = result

for name, result in result_dict.iteritems():
    print "%s = %7.3f" % (name, result)

Вывод

set =   1.711
del =   3.450
pop =   3.618

Таким образом, генератор с индексами в setбыл победителем. И delнемного быстрее, чем тогда pop.

Вы можете использовать эту логику:

my_list = ['word','yes','no','nice']

c=[b for i,b in enumerate(my_list) if not i in (0,2,3)]

print c

Еще одна реализация идеи удаления из высшего индекса.

for i in range(len(yourlist)-1, -1, -1):
    del yourlist(i)

Я действительно могу придумать два способа сделать это:

1. нарезать список как (это удаляет 1-й, 3-й и 8-й элементы)

   somelist = somelist [1: 2] + somelist [3: 7] + somelist [8:]

2. сделать это на месте, но по одному:

   somelist.pop (2) somelist.pop (0)

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

Простой код, чтобы объяснить это , выполнив :

>>> lst = ['a','b','c']
>>> dct = {0: 'a', 1: 'b', 2:'c'}
>>> lst[0]
'a'
>>> dct[0]
'a'
>>> del lst[0]
>>> del dct[0]
>>> lst[0]
'b'
>>> dct[0]
Traceback (most recent call last):
  File "<pyshell#19>", line 1, in <module>
    dct[0]
KeyError: 0
>>> dct[1]
'b'
>>> lst[1]
'c'

Способ "преобразовать" список в dict:

>>> dct = {}
>>> for i in xrange(0,len(lst)): dct[i] = lst[i]

Обратное:

lst = [dct[i] for i in sorted(dct.keys())] 

В любом случае, я думаю, что лучше начинать удаление с более высокого индекса, как вы сказали.

Обобщить комментарий от @sth . Удаление элемента в любом классе, который реализует abc.MutableSequence , и listв частности, выполняется с помощью __delitem__магического метода. Этот метод работает аналогично __getitem__, то есть он может принимать целое число или фрагмент. Вот пример:

class MyList(list):
    def __delitem__(self, item):
        if isinstance(item, slice):
            for i in range(*item.indices(len(self))):
                self[i] = 'null'
        else:
            self[item] = 'null'


l = MyList(range(10))
print(l)
del l[5:8]
print(l)

Это будет выводить

[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
[0, 1, 2, 3, 4, 'null', 'null', 'null', 8, 9]

Импортировать его только по этой причине может быть излишним, но если вы pandasвсе равно используете его , то решение простое и понятное:

import pandas as pd
stuff = pd.Series(['a','b','a','c','a','d'])
less_stuff = stuff[stuff != 'a']  # define any condition here
# results ['b','c','d']

some_list.remove(some_list[max(i, j)])

Позволяет избежать стоимости сортировки и необходимости явно копировать список.

Как насчет одного из них (я очень плохо знаком с Python, но они кажутся нормальными):

ocean_basin = ['a', 'Atlantic', 'Pacific', 'Indian', 'a', 'a', 'a']
for i in range(1, (ocean_basin.count('a') + 1)):
    ocean_basin.remove('a')
print(ocean_basin)

[«Атлантика», «Тихоокеанский», «Индийский»]

ob = ['a', 'b', 4, 5,'Atlantic', 'Pacific', 'Indian', 'a', 'a', 4, 'a']
remove = ('a', 'b', 4, 5)
ob = [i for i in ob if i not in (remove)]
print(ob)

[«Атлантика», «Тихоокеанский», «Индийский»]

Ни один из предложенных ответов пока не выполняет удаление в месте O (n) по длине списка для произвольного числа индексов, которые нужно удалить, поэтому вот моя версия:

def multi_delete(the_list, indices):
    assert type(indices) in {set, frozenset}, "indices must be a set or frozenset"
    offset = 0
    for i in range(len(the_list)):
        if i in indices:
            offset += 1
        elif offset:
            the_list[i - offset] = the_list[i]
    if offset:
        del the_list[-offset:]

# Example:
a = [0, 1, 2, 3, 4, 5, 6, 7]
multi_delete(a, {1, 2, 4, 6, 7})
print(a)  # prints [0, 3, 5]

Вы также можете использовать удалить.

delete_from_somelist = []
for i in [int(0), int(2)]:
     delete_from_somelist.append(somelist[i])
for j in delete_from_somelist:
     newlist = somelist.remove(j)

Я собрал все это вместе в list_diffфункцию, которая просто принимает два списка в качестве входных данных и возвращает их различие, сохраняя при этом первоначальный порядок первого списка.

def list_diff(list_a, list_b, verbose=False):

    # returns a difference of list_a and list_b,
    # preserving the original order, unlike set-based solutions

    # get indices of elements to be excluded from list_a
    excl_ind = [i for i, x in enumerate(list_a) if x in list_b]
    if verbose:
        print(excl_ind)

    # filter out the excluded indices, producing a new list 
    new_list = [i for i in list_a if list_a.index(i) not in excl_ind]
    if verbose:
        print(new_list)

    return(new_list)

Пример использования:

my_list = ['a', 'b', 'c', 'd', 'e', 'f', 'woof']
# index = [0, 3, 6]

# define excluded names list
excl_names_list = ['woof', 'c']

list_diff(my_list, excl_names_list)
>> ['a', 'b', 'd', 'e', 'f']