Индекс все * кроме * один элемент в python

Есть ли простой способ индексировать все элементы списка (или массив или что-то еще) кроме для определенного индекса? Например,

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

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

Ответ 1

Для списка вы можете использовать список comp. Например, чтобы сделать b копию a без третьего элемента:

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]

Это очень общее и может использоваться со всеми повторами, включая массивы numpy. Если вы замените [] на (), 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])

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

Ответ 3

Самый простой способ я нашел:

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

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

Ответ 4

Если вы используете 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]

Ответ 5

Используйте 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]])

Ответ 6

Если вы не знаете, что индекс заранее здесь, это функция, которая будет работать

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