Comment extraire des éléments d’une liste basée sur des indices connus ?

Récupération de plusieurs éléments d'une liste basée sur des indices connus

Pour extraire des éléments spécifiques d'une liste connaissant leurs indices, vous pouvez utiliser les éléments suivants méthodes :

1. Compréhension de liste :

Comme vous l'avez déjà démontré, les compréhensions de liste fournissent un moyen concis d'extraire des éléments en fonction d'indices :

a = [-2, 1, 5, 3, 8, 5, 6]
b = [1, 2, 5]
c = [a[i] for i in b]

2. Operator.itemgetter :

L'opérateur itemgetter du module opérateur permet de récupérer des éléments en fonction d'une séquence d'indices :

from operator import itemgetter
a = [-2, 1, 5, 3, 8, 5, 6]
b = [1, 2, 5]
print(itemgetter(*b)(a))
# Result: (1, 5, 5)

3. Numpy (si disponible) :

Si Numpy est installé, vous pouvez tirer parti de ses capacités d'indexation de tableaux :

import numpy as np
a = np.array([-2, 1, 5, 3, 8, 5, 6])
b = [1, 2, 5]
print(list(a[b]))
# Result: [1, 5, 5]

Bien que les trois méthodes puissent accomplir la tâche, la compréhension de la liste Cette approche est généralement considérée comme la plus lisible et la plus concise. Cependant, si vous appréciez la vitesse ou l'efficacité, itemgetter ou Numpy peuvent offrir de meilleures performances.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!