Python – suma de dígitos K

Dada una lista numérica, extraiga la suma del K-ésimo dígito.

Entrada : test_list = [5467, 34232, 45456, 22222, 3455], K = 2 
Salida : 19 
Explicación : 6 + 2 + 4 + 2 + 5 = 19.

Entrada : test_list = [5467, 34232, 45456, 22222, 3455], K = 0 
Salida : 17 
Explicación : 5 + 3 + 4 + 2 + 3 = 17. 

Método #1: Usar str() + bucle

En esto, convertimos el elemento en una string y luego calculamos la suma de solo el K-ésimo dígito extrayéndolo usando un bucle.

# Python3 code to demonstrate working of
# Kth digit Sum
# Using loop + sum()
  
# initializing list
test_list = [5467, 34232, 45456, 22222, 3455]
  
# printing original list
print("The original list is : " + str(test_list))
  
# initializing K
K = 2
  
res = 0
for ele in test_list:
  
    # adding Kth digit
    res += int(str(ele)[K])
  
# printing result
print("Kth digit sum : " + str(res))

Producción:

The original list is : [5467, 34232, 45456, 22222, 3455]
Kth digit sum : 19

Método #2: Usando sum() + comprensión de lista + str()

En esto, realizamos la tarea de obtener la suma usando sum(), y la comprensión de la lista se usa para obtener un enfoque sencillo del problema. 

# Python3 code to demonstrate working of 
# Kth digit Sum 
# Using sum() + list comprehension + str()
  
# initializing list
test_list = [5467, 34232, 45456, 22222, 3455]
  
# printing original list
print("The original list is : " + str(test_list))
  
# initializing K 
K = 2
  
# sum() getting summation
res = sum([int(str(ele)[K]) for ele in test_list])
  
# printing result 
print("Kth digit sum : " + str(res))

Producción:

The original list is : [5467, 34232, 45456, 22222, 3455]
Kth digit sum : 19