trovare e sostituire elementi in un elenco

Devo cercare in un elenco e sostituire tutte le occorrenze di un elemento con un altro. Finora i miei tentativi nel codice non mi stanno portando da nessuna parte, qual è il modo migliore per farlo?

Ad esempio, supponiamo che la mia lista abbia i seguenti numeri interi

>>> a = [1,2,3,4,5,1,2,3,4,5,1]

e devo sostituire tutte le occorrenze del numero 1 con il valore 10, quindi l'output di cui ho bisogno è

>>> a = [10, 2, 3, 4, 5, 10, 2, 3, 4, 5, 10]

Quindi il mio obiettivo è sostituire tutte le istanze del numero 1 con il numero 10.

>>> a= [1, 2, 3, 4, 5, 1, 2, 3, 4, 5, 1]
>>> for n, i in enumerate(a):
...   if i == 1:
...      a[n] = 10
...
>>> a
[10, 2, 3, 4, 5, 10, 2, 3, 4, 5, 10]

Prova a usare una comprensione dell'elenco e l' operatore ternario .

>>> a=[1,2,3,1,3,2,1,1]
>>> [4 if x==1 else x for x in a]
[4, 2, 3, 4, 3, 2, 4, 4]

La comprensione dell'elenco funziona bene e il ciclo con enumerate può farti risparmiare un po 'di memoria (b / c l'operazione essenzialmente viene eseguita sul posto).

C'è anche una programmazione funzionale. Vedi l'utilizzo della mappa :

>>> a = [1,2,3,2,3,4,3,5,6,6,5,4,5,4,3,4,3,2,1]
>>> map(lambda x: x if x != 4 else 'sss', a)
[1, 2, 3, 2, 3, 'sss', 3, 5, 6, 6, 5, 'sss', 5, 'sss', 3, 'sss', 3, 2, 1]

Se hai diversi valori da sostituire, puoi anche usare un dizionario:

a = [1, 2, 3, 4, 1, 5, 3, 2, 6, 1, 1]
dic = {1:10, 2:20, 3:'foo'}

print([dic.get(n, n) for n in a])

> [10, 20, 'foo', 4, 10, 5, 'foo', 20, 6, 10, 10]

>>> a=[1,2,3,4,5,1,2,3,4,5,1]
>>> item_to_replace = 1
>>> replacement_value = 6
>>> indices_to_replace = [i for i,x in enumerate(a) if x==item_to_replace]
>>> indices_to_replace
[0, 5, 10]
>>> for i in indices_to_replace:
...     a[i] = replacement_value
... 
>>> a
[6, 2, 3, 4, 5, 6, 2, 3, 4, 5, 6]
>>> 

a = [1,2,3,4,5,1,2,3,4,5,1,12]
for i in range (len(a)):
    if a[i]==2:
        a[i]=123

Puoi usare un ciclo for e oppure while; tuttavia se si conosce la funzione Enumerate integrata, si consiglia di utilizzare Enumerate. 1

Per sostituire facilmente tutti 1con 10in a = [1,2,3,4,5,1,2,3,4,5,1]uno, è possibile utilizzare la seguente combinazione lambda + mappa a una riga e "Guarda, mamma, niente IF o FOR!" :

# This substitutes all '1' with '10' in list 'a' and places result in list 'c':

c = list(map(lambda b: b.replace("1","10"), a))

Il seguente è un metodo molto diretto in Python 2.x

 a = [1,2,3,4,5,1,2,3,4,5,1]        #Replacing every 1 with 10
 for i in xrange(len(a)):
   if a[i] == 1:
     a[i] = 10  
 print a

Questo metodo funziona. I commenti sono ben accetti Spero che sia d'aiuto :)

Anche provare a capire come Outis di e di damzam soluzioni di lavoro. Le compressioni dell'elenco e la funzione lambda sono strumenti utili.

Su elenchi lunghi e eventi rari è circa 3 volte più veloce utilizzando list.index()- rispetto ai metodi di iterazione a passaggio singolo presentati nelle altre risposte.

def list_replace(lst, old=1, new=10):
    """replace list elements (inplace)"""
    i = -1
    try:
        while 1:
            i = lst.index(old, i + 1)
            lst[i] = new
    except ValueError:
        pass

So che questa è una domanda molto antica e ci sono una miriade di modi per farlo. Quello più semplice che ho trovato sta usando il numpypacchetto.

import numpy

arr = numpy.asarray([1, 6, 1, 9, 8])
arr[ arr == 8 ] = 0 # change all occurrences of 8 by 0
print(arr)

Il mio caso d'uso stava sostituendo Nonecon un valore predefinito.

Ho cronometrato gli approcci a questo problema che sono stati presentati qui, incluso quello di @kxr - using str.count.

Codice di prova in ipython con Python 3.8.1:

def rep1(lst, replacer = 0):
    ''' List comprehension, new list '''

    return [item if item is not None else replacer for item in lst]


def rep2(lst, replacer = 0):
    ''' List comprehension, in-place '''    
    lst[:] =  [item if item is not None else replacer for item in lst]

    return lst


def rep3(lst, replacer = 0):
    ''' enumerate() with comparison - in-place '''
    for idx, item in enumerate(lst):
        if item is None:
            lst[idx] = replacer

    return lst


def rep4(lst, replacer = 0):
    ''' Using str.index + Exception, in-place '''

    idx = -1
    # none_amount = lst.count(None)
    while True:
        try:
            idx = lst.index(None, idx+1)
        except ValueError:
            break
        else:
            lst[idx] = replacer

    return lst


def rep5(lst, replacer = 0):
    ''' Using str.index + str.count, in-place '''

    idx = -1
    for _ in range(lst.count(None)):
        idx = lst.index(None, idx+1)
        lst[idx] = replacer

    return lst


def rep6(lst, replacer = 0):
    ''' Using map, return map iterator '''

    return map(lambda item: item if item is not None else replacer, lst)


def rep7(lst, replacer = 0):
    ''' Using map, return new list '''

    return list(map(lambda item: item if item is not None else replacer, lst))


lst = [5]*10**6
# lst = [None]*10**6

%timeit rep1(lst)    
%timeit rep2(lst)    
%timeit rep3(lst)    
%timeit rep4(lst)    
%timeit rep5(lst)    
%timeit rep6(lst)    
%timeit rep7(lst)    

Ottengo:

26.3 ms ± 163 µs per loop (mean ± std. dev. of 7 runs, 10 loops each)
29.3 ms ± 206 µs per loop (mean ± std. dev. of 7 runs, 10 loops each)
33.8 ms ± 191 µs per loop (mean ± std. dev. of 7 runs, 10 loops each)
11.9 ms ± 37.8 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)
11.9 ms ± 60.2 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)
260 ns ± 1.84 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)
56.5 ms ± 204 µs per loop (mean ± std. dev. of 7 runs, 10 loops each)

L'uso dell'interno str.indexè infatti più veloce di qualsiasi confronto manuale.

Non sapevo se l'eccezione nel test 4 sarebbe stata più laboriosa dell'uso str.count, la differenza sembra trascurabile.

Si noti che map()(test 6) restituisce un iteratore e non un elenco effettivo, quindi test 7.

Puoi semplicemente usare la comprensione dell'elenco in Python:

def replace_element(YOUR_LIST, set_to=NEW_VALUE):
    return [i
            if SOME_CONDITION
            else NEW_VALUE
            for i in YOUR_LIST]

nel tuo caso, in cui desideri sostituire tutte le occorrenze di 1 con 10, lo snippet di codice sarà così:

def replace_element(YOUR_LIST, set_to=10):
    return [i
            if i != 1  # keeps all elements not equal to one
            else set_to  # replaces 1 with 10
            for i in YOUR_LIST]

Trova e sostituisci un solo oggetto

ur_list = [1,2,1]     # replace the first 1 wiz 11

loc = ur_list.index(1)
ur_list.remove(1)
ur_list.insert(loc, 11)

----------
[11,2,1]