Індексуйте всі * крім * одного елемента в python

Чи є простий спосіб індексувати всі елементи списку (або масиву чи будь-якого іншого), крім конкретного індексу? Наприклад,

• mylist[3] поверне елемент у позицію 3

• milist[~3] поверне весь список, крім 3

Для списку ви можете скористатися списком комп. Наприклад, щоб зробити bкопію aбез 3-го елемента:

a = range(10)[::-1]                       # [9, 8, 7, 6, 5, 4, 3, 2, 1, 0]
b = [x for i,x in enumerate(a) if i!=3]   # [9, 8, 7, 5, 4, 3, 2, 1, 0]

Це дуже загально, і його можна використовувати з усіма ітерабелями, включаючи нумеровані масиви. Якщо ви заміните []на (), bзамість списку буде ітератор.

Або ви могли це зробити на місці за допомогою pop:

a = range(10)[::-1]     # a = [9, 8, 7, 6, 5, 4, 3, 2, 1, 0]
a.pop(3)                # a = [9, 8, 7, 5, 4, 3, 2, 1, 0]

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

a = np.arange(9, -1, -1)     # a = array([9, 8, 7, 6, 5, 4, 3, 2, 1, 0])
b = a[np.arange(len(a))!=3]  # b = array([9, 8, 7, 5, 4, 3, 2, 1, 0])

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

>>> l = range(1,10)
>>> l
[1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> l[:2] 
[1, 2]
>>> l[3:]
[4, 5, 6, 7, 8, 9]
>>> l[:2] + l[3:]
[1, 2, 4, 5, 6, 7, 8, 9]
>>> 

Дивитися також

Поясніть позначення фрагмента Python

Найпростіший спосіб, який я знайшов:

mylist[:x]+mylist[x+1:]

який створить ваш mylistбез елемента в індексі x.

Якщо ви використовуєте numpy, найближчий, я можу подумати, це використовувати маску

>>> import numpy as np
>>> arr = np.arange(1,10)
>>> mask = np.ones(arr.shape,dtype=bool)
>>> mask[5]=0
>>> arr[mask]
array([1, 2, 3, 4, 5, 7, 8, 9])

Щось подібного можна досягти, використовуючи itertoolsбезnumpy

>>> from itertools import compress
>>> arr = range(1,10)
>>> mask = [1]*len(arr)
>>> mask[5]=0
>>> list(compress(arr,mask))
[1, 2, 3, 4, 5, 7, 8, 9]

Використовуйте np.delete! Це фактично нічого не видаляє на місці

Приклад:

import numpy as np
a = np.array([[1,4],[5,7],[3,1]])                                       

# a: array([[1, 4],
#           [5, 7],
#           [3, 1]])

ind = np.array([0,1])                                                   

# ind: array([0, 1])

# a[ind]: array([[1, 4],
#                [5, 7]])

all_except_index = np.delete(a, ind, axis=0)                                              
# all_except_index: array([[3, 1]])

# a: (still the same): array([[1, 4],
#                             [5, 7],
#                             [3, 1]])

Я збираюся запропонувати функціональний (незмінний) спосіб зробити це.

1. Стандартний і простий спосіб зробити це - використовувати нарізки:

   index_to_remove = 3
   data = [*range(5)]
   new_data = data[:index_to_remove] + data[index_to_remove + 1:]
   
   print(f"data: {data}, new_data: {new_data}")

   Вихід:

   data: [0, 1, 2, 3, 4], new_data: [0, 1, 2, 4]

2. Використання списку розуміння:

   data = [*range(5)]
   new_data = [v for i, v in enumerate(data) if i != index_to_remove]
   
   print(f"data: {data}, new_data: {new_data}") 

   Вихід:

   data: [0, 1, 2, 3, 4], new_data: [0, 1, 2, 4]

3. Використовуйте функцію фільтра:

   index_to_remove = 3
   data = [*range(5)]
   new_data = [*filter(lambda i: i != index_to_remove, data)]

   Вихід:

   data: [0, 1, 2, 3, 4], new_data: [0, 1, 2, 4]

4. Використання маскування. Маскування забезпечується функцією itertools.compress у стандартній бібліотеці:

   from itertools import compress
   
   index_to_remove = 3
   data = [*range(5)]
   mask = [1] * len(data)
   mask[index_to_remove] = 0
   new_data = [*compress(data, mask)]
   
   print(f"data: {data}, mask: {mask}, new_data: {new_data}")

   Вихід:

   data: [0, 1, 2, 3, 4], mask: [1, 1, 1, 0, 1], new_data: [0, 1, 2, 4]

5. Використовуйте функцію itertools.filterfalse зі стандартної бібліотеки Python

   from itertools import filterfalse
   
   index_to_remove = 3
   data = [*range(5)]
   new_data = [*filterfalse(lambda i: i == index_to_remove, data)]
   
   print(f"data: {data}, new_data: {new_data}")

   Вихід:

   data: [0, 1, 2, 3, 4], new_data: [0, 1, 2, 4]

Якщо ви не знаєте індексу заздалегідь, ось функція, яка буде працювати

def reverse_index(l, index):
    try:
        l.pop(index)
        return l
    except IndexError:
        return False

Зауважте, що якщо змінною є список списків, деякі підходи не зможуть. Наприклад:

v1 = [[range(3)] for x in range(4)]
v2 = v1[:3]+v1[4:] # this fails
v2

Для загального випадку використовуйте

removed_index = 1
v1 = [[range(3)] for x in range(4)]
v2 = [x for i,x in enumerate(v1) if x!=removed_index]
v2

Якщо ви хочете вирізати останній чи перший, зробіть це:

list = ["This", "is", "a", "list"]
listnolast = list[:-1]
listnofirst = list[1:]

Якщо ви зміните 1 на 2, перші 2 символи буде видалено не другий. Сподіваюся, це все-таки допомагає!