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

Вы можете использовать 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).

Мне это кажется более читаемым. (Возможно, я бы чувствовал себя иначе, если бы у меня была привычка использовать enumerate.) ПРЕДОСТЕРЕЖЕНИЕ: Я не тестировал производительность двух подходов.

# 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 () - не лучшая идея, поскольку он не может обрабатывать повторяющиеся элементы в списке.

Метод Remove приведет к значительному смещению элементов списка. Думаю, лучше сделать копию:

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

Избегает затрат на сортировку и необходимости явно копировать список.

Ни один из предложенных до сих пор ответов не выполняет удаление на месте в 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]

Вы можете просто использовать np.delete:

list_indices = [0, 2]
original_list = [0, 1, 2, 3]
new_list = np.delete(original_list, list_indices)

Вывод

array([1, 3])

Здесь первый аргумент - это исходный список, второй - индекс или список индексов, которые вы хотите удалить.

Существует третий аргумент, который вы можете использовать в случае наличия ndarrays: ось (0 для строк и 1 для столбцов в случае ndarrays).

Как насчет одного из них (я новичок в 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)

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

Я собрал все это вместе в 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']

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

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)