Comment faire pour extraire des éléments d'une liste à l'aide d'indices en Python?

Si vous avez une liste en python, et que vous voulez extraire l'élément indices dire 1, 2 et 5 en une nouvelle liste, comment voulez-vous faire ?

C'est la façon dont je l'ai fait, mais je ne suis pas très satisfait

>>> a
[10, 11, 12, 13, 14, 15]
>>> [x[1] for x in enumerate(a) if x[0] in [1,2,5]]
[11, 12, 15]

une meilleure façon ?

plus, en général, donné un n-uplet avec des indices, des comment utiliser ce tuple pour extraire les éléments correspondants à partir d'une liste, avec éventuellement de la duplication (par exemple n-uplet (1,1,2,1,5) produit [11,11,12,11,15] )

InformationsquelleAutor Stefano Borini | 2010-04-12
  1. 71

    Peut-être utiliser ce:

    [a[i] for i in (1,2,5)]
# [11, 12, 15]
    • btw c'est aussi beaucoup plus rapide. l'extraction de K éléments dans une liste de taille N est de complexité O(K) ici et O(N*K) à l'OP de la solution.
    • ok... merci. C'était sacrément trivial, et il est clair que je suis trop fatigué.
    • Pas de problème. Toute la journée, je peux répondre à une de Stefano Borini question est l'un des (rares, mais les) bonne journée 🙂
    • euh ? ..........
    • Personnellement je n'ai jamais vu Python écrit de cette façon, mais je suis d'accord. C'est assez de génie, et lisible!
    InformationsquelleAutor unutbu
  2. 6

    Je pense que vous êtes à la recherche pour cette:

    elements = [10, 11, 12, 13, 14, 15]
indices = (1,1,2,1,5)

result_list = [elements[i] for i in indices]
    InformationsquelleAutor lugte098
  3. 5

    Essayer

    numbers = range(10, 16)
indices = (1, 1, 2, 1, 5)

result = [numbers[i] for i in indices]
    InformationsquelleAutor Juho Vepsäläinen
  4. 3

    Utiliser Numpy directe de la matrice de l'indexation, comme dans MATLAB, Julia, ... 

    a = [10, 11, 12, 13, 14, 15];
s = [1, 2, 5] ;

import numpy as np
list(np.array(a)[s])
# [11, 12, 15]

    Mieux encore, il suffit de rester avec Numpy tableaux

    a = np.array([10, 11, 12, 13, 14, 15])
a[s]
#array([11, 12, 15])
    InformationsquelleAutor Dmitri
  5. 2

    Limites vérifié:

     [a[index] for index in (1,2,5,20) if 0 <= index < len(a)]
 # [11, 12, 15]
    InformationsquelleAutor Charles Beattie