Додайте SUM значень двох СПИСОКІВ у новий СПИСОК


135

У мене є такі два списки:

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

Тепер я хочу додати елементи з обох цих списків у новий список.

вихід повинен бути

third = [7,9,11,13,15]

Відповіді:


212

Тут zipкорисна функція, яка використовується з розумінням списку.

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

Якщо у вас є список списків (а не лише два списки):

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

1
просто цікаво, як би обробляти zip (), якщо довжина масиву відрізняється? тобто що повертає zip для різної довжини масиву і як це вплине на операцію для x + y
ealeon

7
@ealeon: "блискавка" припиняється, коли вичерпається найкоротший ітерабел. Так що, якщо firstє довжина 10 і secondє довжиною 6, результат проносячись по ітеріруемому буде довжиною 6.
Том

Я думаю, що це корисніше, ніж інші відповіді, тому що ви можете робити такі корисні речі, як середня оцінка або надати різну вагу кожному елементу в масиві та комбінувати їх
seokhoonlee

Чи можна це зробити, коли ви не знаєте кількості списків?
traggatmot

@traggatmot: >>> lists_of_lists = [[1, 2, 3], [4, 5, 6]] >>> [sum(x) for x in zip(*lists_of_lists)] [5, 7, 9]
Tom


27

Якщо припустити обидва списки aтаb мають однакову довжину, вам не потрібно ZIP, NumPy або що - небудь ще.

Python 2.x і 3.x:

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

це здається досить гарним, якщо нам доведеться обчислити підсумковий підсумок для більш ніж 2 списків
lazarus

25

Поведінка за замовчуванням у numpy додається компонентно

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

який виводить

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

На сьогодні найкраща відповідь
Іван,

Це повинно працювати, але в моїх експериментах це не так… Я не знаю чому, але нуме здається мені потужною бібліотекою, а також складною…
decadenza

@decadenza Як ви провели експеримент?
Ашфак

Привіт @Ashfaq, минуло багато місяців, і я краще навчився бібліотеці «Numpy». Я помилявся у визначенні np.array. Вибачте.
декаденса

використовуйте np.add (перший, другий) .tolist (), щоб отримати результат у списку
talekeDskobeDa

11

Це поширюється на будь-яку кількість списків:

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

У вашому випадку myListOfListsбуло б[first, second]


1
Ви впевнені izip.from_iterable?
DSM

1
@DSM: чорт! Я думаю, що я думав chain. Оновлено
inspectorG4dget

9

Спробуйте наступний код:

first = [1, 2, 3, 4]
second = [2, 3, 4, 5]
third = map(sum, zip(first, second))

+1 для цього компактного та зрозумілого рішення. Варто зазначити, що це працює і для більш ніж двох списків:map(sum, zip(first, second, third, fourth, ...))
Йохан Деттмар

6

Найпростіший і швидкий спосіб зробити це:

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

Крім того, ви можете використовувати numpy суму:

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

1
Це гарно узагальнює довші списки списків, що саме те, що мені було потрібно!
Vectornaut



2

Моя відповідь повторюється з тиру, який відповів на неї 17 березня о 9:25.

Це було простіше і швидше, ось його рішення:

Найпростіший і швидкий спосіб зробити це:

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

Крім того, ви можете використовувати numpy суму:

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

Вам нудно!

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]

2

Якщо у вас є невідома кількість списків однакової довжини, ви можете скористатися функцією нижче.

Тут * args приймає змінну кількість аргументів списку (але сумує лише однакову кількість елементів у кожному). * Знову використовується для розпакування елементів у кожному зі списків.

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

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

sum_lists(a,b)

Вихід:

[2, 4, 6]

Або з 3 списками

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

Вихід:

[19, 19, 19, 19, 19]

1

Ви можете використовувати zip(), який "переплете" два масиви разом, а потім map(), який застосує функцію до кожного елемента в ітерабелі:

>>> 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]

1

Ось ще один спосіб зробити це. Ми використовуємо внутрішню __add__ функцію 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)

Вихідні дані

[11, 22, 33, 44, 55]

1

Якщо ви хочете додати також решта значень у списках, ви можете використовувати це (це працює в 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))

1
    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]

1

Ось ще один спосіб зробити це. Це для мене добре.

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("")

1
j = min(len(l1), len(l2))
l3 = [l1[i]+l2[i] for i in range(j)]

1
Хоча цей фрагмент коду може бути вирішенням, зокрема пояснення дійсно допомагає покращити якість вашої публікації. Пам'ятайте, що ви відповідаєте на запитання читачів у майбутньому, і ці люди можуть не знати причини вашої пропозиції щодо коду.
Нарендра Ядхав

1

Мабуть, найпростіший підхід:

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

Якщо ви розглядаєте свої списки як масивний масив, вам потрібно легко підсумувати їх:

import numpy as np

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

print third

[7, 9, 11, 13, 15]

0

Що робити, якщо у вас список різної довжини, ви можете спробувати щось подібне (використовуючи zip_longest)

from itertools import zip_longest  # izip_longest for python2.x

l1 = [1, 2, 3]
l2 = [4, 5, 6, 7]

>>> list(map(sum, zip_longest(l1, l2, fillvalue=0)))
[5, 7, 9, 7]

-2

Ви можете використовувати цей метод, але він буде працювати лише в тому випадку, якщо обидва списку мають однаковий розмір:

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

a = len(first)
b = int(0)
while True:
    x = first[b]
    y = second[b]
    ans = x + y
    third.append(ans)
    b = b + 1
    if b == a:
        break

print third
Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.