Есть ли простой способ индексировать все элементы списка (или массив или что-то еще) кроме для определенного индекса? Например,
-
mylist[3]
вернет элемент в позиции 3 -
milist[~3]
вернет весь список, кроме 3
Есть ли простой способ индексировать все элементы списка (или массив или что-то еще) кроме для определенного индекса? Например,
mylist[3]
вернет элемент в позиции 3
milist[~3]
вернет весь список, кроме 3
Для списка вы можете использовать список 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])
который будет, в общем, намного быстрее, чем перечисленное выше перечисление.
>>> 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]
>>>
См. также
Самый простой способ я нашел:
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]])
Если вы не знаете, что индекс заранее здесь, это функция, которая будет работать
def reverse_index(l, index):
try:
l.pop(index)
return l
except IndexError:
return False