Remplacer des éléments sélectionnés dans une liste en Python

Question

J'ai une liste: mylist = [0, 0, 0, 0, 0]

Je veux seulement remplacer les éléments sélectionnés, disent les première, deuxième et quatrième par un numéro commun, A = 100.

Une façon de le faire:

mylist[:2] = [A]*2 
mylist[3] = A 
mylist 
[100, 100, 0, 100, 0] 

Je cherche une doublure ou une méthode plus facile de le faire. Une réponse plus générale et plus flexible est préférable.

Answer 1

D'autant plus que vous remplacez un morceau considérable de la list, je ferais ce immuablement:

mylist = [100 if i in (0, 1, 3) else e for i, e in enumerate(mylist)] 

Il est intentionnel en Python qui fait une nouvelle list est un one-liner, tout unmuternécessite une boucle explicite. Habituellement, si vous ne savez pas lequel vous voulez, vous voulez le nouveau list. (Dans certains cas, il est plus lent ou plus compliqué, ou vous avez un autre code qui fait référence au même list et doit le voir muté, ou quoi que ce soit, c'est pourquoi c'est "habituellement" plutôt que "toujours".)

Si vous voulez faire plus d'une fois, je l'envelopper dans une fonction, comme la volatilité suggère:

def elements_replaced(lst, new_element, indices): 
    return [new_element if i in indices else e for i, e in enumerate(lst)] 

personnellement, je faire probablement un générateur de sorte qu'il donne une itération au lieu de retourner une liste, même si je n'en aurai jamais besoin, juste parce que je suis stupide de cette façon. Mais si vous avez réellement faire besoin:

myiter = (100 if i in (0, 1, 3) else e for i, e in enumerate(mylist)) 

Ou:

def elements_replaced(lst, new_element, indices): 
    for i, e in enumerate(lst): 
     if i in indices: 
      yield new_element 
     else: 
      yield e 

Answer 2

Est-ce ce que vous cherchez? Faites une liste des index que vous souhaitez modifier, puis parcourez cette liste pour modifier les valeurs.

els_to_replace = [0, 1, 3] 

mylist = [0, 0, 0, 0, 0] 

for index in els_to_replace: 
    mylist[index] = 100 


mylist 
Out[9]: [100, 100, 0, 100, 0] 

Answer 3

def replace_element(lst, new_element, indices): 
    for i in indices: 
     lst[i] = new_element 
    return lst 

Il est certainement une solution plus générale, pas un one-liner bien. Par exemple, dans votre cas, vous appelez:

mylist = replace_element(mylist, 100, [0, 1, 3]) 

Answer 4

Numpy soutient si vous n'êtes pas opposé à l'aide d'un np.ndarray:

>>> a = np.zeros(5) 
>>> a[[0,1,3]] = 100 
>>> a 
array([ 100., 100., 0., 100., 0.]) 

Answer 5

pas un grand fan de celui-ci, mais vous pouvez essayer cela (même si je pense que tous les ci-dessus sont beaucoup plus concis et facile à lire):

In [22]: from operator import setitem 

In [23]: mylist = [0, 0, 0, 0, 0] 

In [24]: indeces_to_replace = [0, 1, 3] 

In [25]: _ = map(lambda x: setitem(mylist, x, 100), indeces_to_replace) 

In [26]: mylist 
Out[26]: [100, 100, 0, 100, 0] 

Mis à part la lisibilité douteuse et la nécessité d'une importation, @abarnert a souligné quelques questions supplémentaires, à savoir que 012.369.crée toujours une liste inutile (qui est rejetée avec le _ mais créé néanmoins) et que cela ne fonctionnera pas dans Python 3 car mapreturns an iterator dans Python 3.x. Vous pouvez utiliser le module six pour simuler le comportement de map dans Python 3.x à partir de Python 2.x, et en combinaison avec collections.deque (encore une fois comme suggéré par @abarnert), vous pouvez obtenir la même sortie sans créer la liste supplémentaire en mémoire car un deque pouvant contenir un maximum de 0 éléments rejettera tout ce qu'il reçoit de l'itérateur map (notez qu'avec six, map est simulé en utilisant itertools.imap).

Encore une fois, il n'y a absolument pas besoin d'utiliser ce jamais - toutes les solutions ci-dessus/dessous est mieux :)

In [1]: from collections import deque 

In [2]: from six.moves import map 

In [3]: from operator import setitem 

In [4]: mylist = [0, 0, 0, 0, 0] 

In [5]: indeces_to_replace = [0, 1, 3] 

In [6]: deque(map(lambda x: setitem(mylist, x, 100), indeces_to_replace), maxlen=0) 
Out[6]: deque([], maxlen=0) 

In [7]: mylist 
Out[7]: [100, 100, 0, 100, 0] 

Answer 6

J'aime une compréhension de la liste:

[100 if index in [1, 4] else 0 for index, x in enumerate(mylist) ]