Indicizza tutto * tranne * un elemento in python


112

Esiste un modo semplice per indicizzare tutti gli elementi di un elenco (o array, o altro) tranne un particolare indice? Per esempio,

  • mylist[3] restituirà l'articolo nella posizione 3

  • milist[~3] restituirà l'intero elenco tranne 3

Risposte:


111

Per una lista , potresti usare una lista comp. Ad esempio, per creare buna copia di asenza il 3 ° elemento:

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]

Questo è molto generale e può essere utilizzato con tutti gli iterabili, inclusi gli array numpy. Se sostituisci []con (), bsarà un iteratore invece di un elenco.

Oppure potresti farlo sul posto con 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]

In numpy potresti farlo con un'indicizzazione booleana:

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

che sarà, in generale, molto più veloce della comprensione della lista sopra elencata.


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

Guarda anche

Spiega la notazione slice di Python


2
Buona risposta per una lista. Potresti usarlo anche per gli array, ma dovresti usarlo numpy.concatenate.
Bi Rico

48

Il modo più semplice che ho trovato è stato:

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

che produrrà il tuo mylistsenza l'elemento in index x.


Ho trovato questo in realtà rimuovere l'elemento x + 1, ancora utile però, grazie
Jack TC

24

Se stai usando numpy, il più vicino, posso pensare è usare una maschera

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

Qualcosa di simile può essere ottenuto usando itertoolssenzanumpy

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

2
Potrei usare qualcosa di simile np.arange(len(arr)) != 3come maschera, perché poi può essere inline, ad esempio arr[~(np.arange(len(arr)) == 3)]o qualsiasi altra cosa.
DSM

@DSM: per favore pubblica questo come risposta :-). In ogni caso, non ho molta dimestichezza con Numpy.
Abhijit

+1 per mascherare un array, nel caso di un elenco andrei con slice e concatenerei usando compress.
Bi Rico

5

Usa np.delete! In realtà non elimina nulla al suo posto

Esempio:

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

2

Fornirò un modo funzionale (immutabile) per farlo.

  1. Il modo standard e semplice per farlo è utilizzare l'affettatura:

    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}")

    Produzione:

    data: [0, 1, 2, 3, 4], new_data: [0, 1, 2, 4]
  2. Usa la comprensione dell'elenco:

    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}") 

    Produzione:

    data: [0, 1, 2, 3, 4], new_data: [0, 1, 2, 4]
  3. Usa la funzione di filtro:

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

    Produzione:

    data: [0, 1, 2, 3, 4], new_data: [0, 1, 2, 4]
  4. Utilizzo del mascheramento. Il mascheramento è fornito dalla funzione itertools.compress nella libreria standard:

    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}")

    Produzione:

    data: [0, 1, 2, 3, 4], mask: [1, 1, 1, 0, 1], new_data: [0, 1, 2, 4]
  5. Usa la funzione itertools.filterfalse dalla libreria standard di 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}")

    Produzione:

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

0

Se non conosci l'indice in anticipo, ecco una funzione che funzionerà

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

0

Nota che se la variabile è un elenco di elenchi, alcuni approcci fallirebbero. Per esempio:

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

Per il caso generale, usa

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

0

Se vuoi ritagliare l'ultimo o il primo, fai questo:

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

Se si cambia da 1 a 2, i primi 2 caratteri verranno rimossi, non il secondo. Spero che questo aiuti ancora!

Utilizzando il nostro sito, riconosci di aver letto e compreso le nostre Informativa sui cookie e Informativa sulla privacy.
Licensed under cc by-sa 3.0 with attribution required.