A veces, nos encontramos con tal problema en el que necesitamos encontrar la suma de cada columna en una array, es decir, la suma de cada índice en la lista de listas. Este tipo de problema es bastante común y útil en la programación competitiva. Analicemos ciertas formas en que se puede resolver este problema.
Método n.º 1:sum() + list comprehension + zip()
se requiere la combinación de los métodos anteriores para resolver este problema en particular. La función de suma se usa para obtener el valor de suma requerido y la función zip proporciona la combinación de índices similares y luego se crea la lista usando la comprensión de lista.
# Python3 code to demonstrate
# Summation of each column in Matrix
# using sum() + list comprehension + zip()
# initializing list
test_list = [[3, 7, 6], [1, 3, 5], [9, 3, 2]]
# printing original list
print("The original list : " + str(test_list))
# using sum() + list comprehension + zip()
# Summation of each column in Matrix
res = [sum(idx) for idx in zip(*test_list)]
# print result
print("The Summation of each index list is : " + str(res))
The original list : [[3, 7, 6], [1, 3, 5], [9, 3, 2]] The Summation of each index list is : [13, 13, 13]
Método n.º 2: Usarmap() + sum() + zip()
Esto funciona casi de manera similar al método anterior, pero la diferencia es que usamos la función de mapa para construir la lista de suma en lugar de usar la comprensión de lista.
# Python3 code to demonstrate
# Summation of each column in Matrix
# using map() + sum() + zip()
# initializing list
test_list = [[3, 7, 6], [1, 3, 5], [9, 3, 2]]
# printing original list
print("The original list : " + str(test_list))
# using map() + sum() + zip()
# Summation of each column in Matrix
res = list(map(sum, zip(*test_list)))
# print result
print("The Summation of each index list is : " + str(res))
The original list : [[3, 7, 6], [1, 3, 5], [9, 3, 2]] The Summation of each index list is : [13, 13, 13]
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