A veces, podemos encontrarnos con una utilidad en la que necesitamos obtener los primeros N elementos de la sublista que también son solo un índice en particular. Esto puede tener una aplicación en cola para obtener solo el nombre de la N persona inicial. Vamos a discutir ciertas formas en que esto se puede hacer.
Método n.º 1: usar la comprensión de listas y el corte
de listas Las dos potentes utilidades de Python anteriores también pueden ser útiles aquí para obtener el resultado, ya que la comprensión de listas puede extraer el corte de elementos y puede restringir el tamaño que necesitamos extraer.
# Python3 code to demonstrate
# getting sublist element till N
# using list comprehension + list slicing
# initializing list
test_list = [['Geeks', 1, 15], ['for', 3, 5], ['Geeks', 3, 7]]
# printing original list
print("The original list : " + str(test_list))
# initializing N
N = 2
# using list comprehension + list slicing
# getting sublist element till N
res = [i[0] for i in test_list[ : N]]
# print result
print("The first element of sublist till N : " + str(res))
The original list : [['Geeks', 1, 15], ['for', 3, 5], ['Geeks', 3, 7]] The first element of sublist till N : ['Geeks', 'for']
Método #2: Usarmap() + itemgetter() + islice()
La combinación de las 3 funciones anteriores se puede usar para realizar esta tarea en particular. La función itemgetter obtiene el elemento a extraer, lo corta en rodajas hasta N y la función map combina el resultado.
# Python3 code to demonstrate
# getting sublist element till N
# using map() + itemgetter() + islice()
from operator import itemgetter
from itertools import islice
# initializing list
test_list = [['Geeks', 1, 15], ['for', 3, 5], ['Geeks', 3, 7]]
# printing original list
print("The original list : " + str(test_list))
# initializing N
N = 2
# using map() + itemgetter() + islice()
# getting sublist element till N
res = list(map(itemgetter(0), islice(test_list, 0, N)))
# print result
print("The first element of sublist till N : " + str(res))
The original list : [['Geeks', 1, 15], ['for', 3, 5], ['Geeks', 3, 7]] The first element of sublist till N : ['Geeks', 'for']
Publicación traducida automáticamente
Artículo escrito por manjeet_04 y traducido por Barcelona Geeks. The original can be accessed here. Licence: CCBY-SA