Comment extraire efficacement des éléments d'une liste basée sur des indices ?

Accès à plusieurs éléments de liste par index

Pour extraire des éléments spécifiques d'une liste en fonction de leurs indices, différentes approches sont disponibles. Prenons l'exemple de l'extraction d'éléments avec les indices 1, 2 et 5 de la liste [-2, 1, 5, 3, 8, 5, 6].

Utilisation d'une compréhension de liste

Comme démontré dans la question, une compréhension de liste peut être utilisée efficacement :

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

Cette approche parcourt efficacement la liste d'index b et extrait les éléments correspondants de la liste a.

Utilisation de Operator.itemgetter

La fonction Operator.itemgetter de Python fournit un aperçu concis solution :

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

Utilisation de NumPy

Pour ceux qui utilisent NumPy, il propose une approche alternative :

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

Considérations relatives aux performances

Bien que le choix dépende de facteurs tels que la taille de la liste et En fonction de la fréquence de son utilisation, la compréhension de liste fournie fonctionne généralement bien pour les petites listes. Cependant, si la liste est longue ou si elle est fréquemment consultée, itemgetter ou NumPy peuvent être préférables en raison de leurs performances améliorées.

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!