web-dev-qa-db-fra.com

Ajouter la somme des valeurs de deux listes dans la nouvelle liste

J'ai les deux listes suivantes:

first = [1,2,3,4,5]
second = [6,7,8,9,10]

Maintenant, je veux ajouter les éléments de ces deux listes dans une nouvelle liste.

la sortie devrait être

third = [7,9,11,13,15]
pythonlistsum
115
Prashant Gaur

La fonction Zip est utile ici, utilisée avec une compréhension en liste.

[x + y for x, y in Zip(first, second)]

Si vous avez une liste de listes (au lieu de deux listes seulement):

lists_of_lists = [[1, 2, 3], [4, 5, 6]]
[sum(x) for x in Zip(*lists_of_lists)]
# -> [5, 7, 9]
172
tom

De docs

import operator
list(map(operator.add, first,second))
47
Thai Tran

En supposant que les deux listes a et b aient la même longueur, vous n'avez pas besoin de Zip, de Numpy ou de quoi que ce soit d'autre.

Python 2.x et 3.x:

[a[i]+b[i] for i in range(len(a))]
27
math

Le comportement par défaut dans numpy est add composantwise

import numpy as np
np.add(first, second)

qui sort

array([7,9,11,13,15])
19
user3582790

Cela s’étend à un nombre quelconque de listes:

[sum(sublist) for sublist in itertools.izip(*myListOfLists)]

Dans votre cas, myListOfLists serait [first, second]

11
inspectorG4dget

Essayez le code suivant:

first = [1, 2, 3, 4]
second = [2, 3, 4, 5]
third = map(sum, Zip(first, second))
9
Maciej Puczkowski
first = [1, 2, 3, 4, 5]
second = [6, 7, 8, 9, 10]
three = map(lambda x,y: x+y,first,second)
print three



Output 
[7, 9, 11, 13, 15]
5
Anurag Misra

Le moyen facile et rapide de le faire est:

three = [sum(i) for i in Zip(first,second)] # [7,9,11,13,15]

Alternativement, vous pouvez utiliser numpy sum:

from numpy import sum
three = sum([first,second], axis=0) # array([7,9,11,13,15])
5
Thiru

Peut-être l'approche la plus simple:

first = [1,2,3,4,5]
second = [6,7,8,9,10]
three=[]

for i in range(0,5):
    three.append(first[i]+second[i])

print(three)
1
Rishabh Kumar

Voici une autre façon de le faire. Nous utilisons la fonction interne __add__ de python:

class SumList(object):
    def __init__(self, this_list):
        self.mylist = this_list

    def __add__(self, other):
        new_list = []
        zipped_list = Zip(self.mylist, other.mylist)
        for item in zipped_list:
            new_list.append(item[0] + item[1])
        return SumList(new_list)

    def __repr__(self):
        return str(self.mylist)

list1 = SumList([1,2,3,4,5])
list2 = SumList([10,20,30,40,50])
sum_list1_list2 = list1 + list2
print(sum_list1_list2)

sortie

[11, 22, 33, 44, 55]
1
Stryker

Ma réponse est répétée avec Thiru qui y a répondu le 17 mars à 9h25.

C'était plus simple et plus rapide, voici ses solutions:

Le moyen facile et rapide de le faire est:

 three = [sum(i) for i in Zip(first,second)] # [7,9,11,13,15]

Alternativement, vous pouvez utiliser numpy sum:

 from numpy import sum
 three = sum([first,second], axis=0) # array([7,9,11,13,15])

Vous avez besoin de numpy!

import numpy as np
a = [1,2,3,4,5]
b = [6,7,8,9,10]
c = list(np.array(a) + np.array(b))
print c
# [7, 9, 11, 13, 15]
0
Piece

Si vous avez un nombre inconnu de listes de même longueur, vous pouvez utiliser la fonction ci-dessous.

Ici, l'argument * accepte un nombre variable d'arguments de liste (mais ajoute seulement le même nombre d'éléments dans chacun). Le * est à nouveau utilisé pour décompresser les éléments dans chacune des listes.

def sum_lists(*args):
    return list(map(sum, Zip(*args)))

a = [1,2,3]
b = [1,2,3]  

sum_lists(a,b)

Sortie:

[2, 4, 6]

Ou avec 3 listes

sum_lists([5,5,5,5,5], [10,10,10,10,10], [4,4,4,4,4])

Sortie:

[19, 19, 19, 19, 19]
0
Wing

Vous pouvez utiliser Zip(), qui "entrelacera" les deux tableaux, puis map(), qui appliquera une fonction à chaque élément d'une variable itérable:

>>> a = [1,2,3,4,5]
>>> b = [6,7,8,9,10]
>>> Zip(a, b)
[(1, 6), (2, 7), (3, 8), (4, 9), (5, 10)]
>>> map(lambda x: x[0] + x[1], Zip(a, b))
[7, 9, 11, 13, 15]
0
cdhowie

Si vous considérez vos listes comme un tableau numpy, vous devez les résumer facilement:

import numpy as np

third = np.array(first) + np.array(second)

print third

[7, 9, 11, 13, 15]
0
Radvin

Si vous voulez aussi ajouter le reste des valeurs dans les listes, vous pouvez utiliser ceci (cela fonctionne dans Python3.5)

def addVectors(v1, v2):
    sum = [x + y for x, y in Zip(v1, v2)]
    if not len(v1) >= len(v2):
        sum += v2[len(v1):]
    else:
        sum += v1[len(v2):]

    return sum


#for testing 
if __name__=='__main__':
    a = [1, 2]
    b = [1, 2, 3, 4]
    print(a)
    print(b)
    print(addVectors(a,b))
0
christianAV

solution monocouche

list(map(lambda x,y: x+y, a,b))
0
Shadowman

Voici une autre façon de le faire. Cela fonctionne très bien pour moi.

N=int(input())
num1 = list(map(int, input().split()))
num2 = list(map(int, input().split()))
sum=[]

for i in range(0,N):
  sum.append(num1[i]+num2[i])

for element in sum:
  print(element, end=" ")

print("")
0
Sidharth yadav
j = min(len(l1), len(l2))
l3 = [l1[i]+l2[i] for i in range(j)]
0
Vijay Kumar
    first = [1,2,3,4,5]
    second = [6,7,8,9,10]
    #one way
    third = [x + y for x, y in Zip(first, second)]
    print("third" , third) 
    #otherway
    fourth = []
    for i,j in Zip(first,second):
        global fourth
        fourth.append(i + j)
    print("fourth" , fourth )
#third [7, 9, 11, 13, 15]
#fourth [7, 9, 11, 13, 15]
0
Bala Srinivasan