Python: extrae los elementos del índice Kth de la lista de valores del diccionario

Dado un diccionario con una lista como valores, extraiga todos los elementos del índice Kth.

Entrada : {“Gfg”: [4, 7, 5], “Mejor”: [8, 6, 7], “es”: [9, 3, 8]}, K = 2
Salida : [5, 7, 8]
Explicación : los elementos del segundo índice son 5, 7 y 8 respectivamente en diferentes claves.

Entrada : {“Gfg”: [4, 7, 5], “Mejor”: [8, 6, 7], “es”: [9, 3, 8]}, K = 0
Salida : [4, 8, 9]
Explicación : los elementos del índice 0 son 4, 8 y 9 respectivamente en diferentes claves.

Método #1: Usar lista de comprensión + valores()

La combinación de las funcionalidades anteriores se puede utilizar para resolver este problema. En esto, los valores se extraen usando valores() y la comprensión de la lista se usa para construir una nueva lista.

# Python3 code to demonstrate working of 
# Extract Kth index elements from Dictionary Value list
# Using list comprehension + values()
  
# initializing dictionary
test_dict = {"Gfg" : [4, 7, 5], "Best" : [8, 6, 7], "is" : [9, 3, 8]}
  
# printing original dictionary
print("The original dictionary is : " + str(test_dict))
  
# initializing K 
K = 1
  
# one liner, values() getting all value according to keys
res = [sub[K] for sub in test_dict.values()]
  
# printing result 
print("The extracted values : " + str(res)) 

The original dictionary is : {'Gfg': [4, 7, 5], 'Best': [8, 6, 7], 'is': [9, 3, 8]}
The extracted values : [7, 6, 3]

Método #2: Usando map() + itemgetter()

La combinación de las funcionalidades anteriores se puede utilizar para resolver este problema. En esto, usamos map() para extender la lógica de obtener valores de una clave particular, y itemgetter se usa para extraer un índice particular.

# Python3 code to demonstrate working of 
# Extract Kth index elements from Dictionary Value list
# Using map() + itemgetter()
from operator import itemgetter
  
# initializing dictionary
test_dict = {"Gfg" : [4, 7, 5], "Best" : [8, 6, 7], "is" : [9, 3, 8]}
  
# printing original dictionary
print("The original dictionary is : " + str(test_dict))
  
# initializing K 
K = 1
  
# map and itemgetter() extracting result 
# list() used to convert result from map() to list format
res = list(map(itemgetter(K), test_dict.values()))
  
# printing result 
print("The extracted values : " + str(res)) 

The original dictionary is : {'Gfg': [4, 7, 5], 'Best': [8, 6, 7], 'is': [9, 3, 8]}
The extracted values : [7, 6, 3]