wake-up-neo.com

Alle indexieren * außer * ein Element in Python

Gibt es eine einfache Möglichkeit, alle Elemente einer Liste (oder eines Arrays oder was auch immer) außer für einen bestimmten Index zu indizieren? Z.B., 

  • mylist[3] gibt den Artikel an Position 3 zurück

  • milist[~3] gibt die gesamte Liste mit Ausnahme von 3 zurück

pythonlistnumpyindexing
67
choldgraf

Für eine Liste können Sie eine Listenkomponente verwenden. B. b eine Kopie von a ohne das 3. Element machen:

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]

Dies ist sehr allgemein und kann mit allen Iterables verwendet werden, einschließlich numpy-Arrays. Wenn Sie [] durch () ersetzen, wird b ein Iterator anstelle einer Liste.

Oder Sie könnten dies direkt mit pop tun:

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]

In numpy können Sie dies mit einer booleschen Indexierung tun:

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

dies ist im Allgemeinen viel schneller als das oben aufgeführte Listenverständnis.

65
tom10
>>> 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]
>>>

Siehe auch

Erklären Sie die Slice-Notation von Python

41
Andreas Jung

Der einfachste Weg, den ich gefunden habe, war:

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

das wird Ihre mylist ohne das Element am Index x erzeugen.

28
Andre Soares

Wenn Sie "numpy" verwenden, kann ich mir am besten vorstellen, eine Maske zu verwenden

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

Etwas Ähnliches kann mit itertools ohne numpy erreicht werden.

>>> 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]
22
Abhijit

Benutze np.delete! Tatsächlich wird nichts an Ort und Stelle gelöscht

Beispiel:

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
brnl

Wenn Sie den Index nicht vorher kennen, ist hier eine Funktion, die funktionieren wird

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

Die obigen Antworten berücksichtigen keinen 0-basierten Ausschluss.

def exclude(lst, i):
    if i == 0:
        return lst[i+1:]

    return lst[:i] + lst[i+1:]
0
Ian