Видалення декількох елементів зі списку

Чи можливо видалити декілька елементів зі списку одночасно? Якщо я хочу видалити елементи в індексах 0 і 2 і спробувати щось на кшталт цього del somelist[0], після чого del somelist[2]другий вислів буде фактично видалений somelist[3].

Я припускаю, що я завжди міг би спочатку видалити більш пронумеровані елементи, але сподіваюся, що є кращий спосіб.

Напевно, не найкраще рішення цієї проблеми:

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 журналу (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).

Мені це здається більш зрозумілим. (Можливо, я відчував би інакше, якби я мав звичку використовувати перерахування.) CAVEAT: Я не перевіряв виконання двох підходів.

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

соліст:

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

соліст:

[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'])

буде recurively видалити foo['bar'], тоfoo['baz']

ми можемо це зробити за допомогою ітерації циклу над індексами після сортування списку індексів у порядку зменшення

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 зі спискуA:

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

Тож генератор з індексами в a 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-й елементи)

   сомеліст = сомеліст [1: 2] + сомеліст [3: 7] + соліст [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'

Спосіб "перетворення" списку в диктат:

>>> 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']