เพิ่ม SUM ของค่าของสอง LISTS ลงใน LIST ใหม่


134

ฉันมีสองรายการต่อไปนี้:

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

ตอนนี้ฉันต้องการเพิ่มรายการจากทั้งสองรายการนี้ลงในรายการใหม่

ผลลัพธ์ควรเป็น

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

คำตอบ:


211

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 () จะจัดการอย่างไรหากความยาวอาร์เรย์แตกต่างกัน? คือสิ่งที่ไม่ซิปผลตอบแทนสำหรับความยาวอาร์เรย์ที่แตกต่างกันและวิธีการที่จะส่งผลกระทบต่อการดำเนินงานว่าสำหรับ 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]
ทอม


26

สมมติทั้งสองรายการaและbมีความยาวเดียวกันคุณไม่จำเป็นต้องซิป numpy หรือสิ่งอื่นใด

Python 2.x และ 3.x:

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

ดูเหมือนจะค่อนข้างดีถ้าเราต้องคำนวณรายการผลรวมผลลัพธ์มากกว่า 2 รายการ
lazarus

24

ลักษณะการทำงานเริ่มต้นใน numpy คือเพิ่ม componentwise

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

ซึ่งเอาต์พุต

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

คำตอบที่ดีที่สุด
เอียน

มันควรจะใช้งานได้ แต่ในการทดลองของฉันมันไม่ได้ ... ฉันไม่รู้ว่าทำไม แต่ดูเหมือนว่าฉันจะเป็นห้องสมุดที่มีพลังเช่นเดียวกับห้องสมุดที่ซับซ้อน ...
decadenza

@decadenza คุณทำการทดลองได้อย่างไร?
Ashfaq

สวัสดี @Ashfaq หลายเดือนผ่านไปฉันได้เรียนรู้ห้องสมุด Numpy ได้ดีขึ้น ฉันผิดในนิยาม np.array ขอโทษ
Decadenza

ใช้ 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 สำหรับโซลูชันขนาดกะทัดรัดและอธิบายตัวเองได้ เป็นที่น่าสังเกตว่ามันใช้งานได้มากกว่า 2 รายการเช่นกัน:map(sum, zip(first, second, third, fourth, ...))
Johan Dettmar

6

วิธีที่ง่ายและรวดเร็วในการดำเนินการนี้คือ:

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

หรือคุณสามารถใช้ numpy sum:

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

1
นี่เป็นการสรุปรายชื่อรายการที่ยาวขึ้นซึ่งเป็นสิ่งที่ฉันต้องการ!
Vectornaut



2

คำตอบของฉันซ้ำกับของ Thiru ที่ตอบในวันที่ 17 มี.ค. เวลา 09:25 น.

มันง่ายขึ้นและเร็วขึ้นนี่คือวิธีแก้ปัญหาของเขา:

วิธีที่ง่ายและรวดเร็วในการดำเนินการนี้คือ:

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

หรือคุณสามารถใช้ numpy sum:

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

คุณต้องมึน!

อาร์เรย์จำนวนนับสามารถดำเนินการบางอย่างเช่นเวกเตอร์

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

นี่เป็นอีกวิธีที่จะทำได้มันทำงานได้ดีสำหรับฉัน

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
แม้ว่าข้อมูลโค้ดนี้อาจเป็นวิธีแก้ปัญหา แต่การมีคำอธิบายจะช่วยปรับปรุงคุณภาพของโพสต์ของคุณได้มาก โปรดจำไว้ว่าคุณกำลังตอบคำถามสำหรับผู้อ่านในอนาคตและบุคคลเหล่านั้นอาจไม่ทราบสาเหตุของการแนะนำโค้ดของคุณ
Narendra Jadhav

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
โดยการใช้ไซต์ของเรา หมายความว่าคุณได้อ่านและทำความเข้าใจนโยบายคุกกี้และนโยบายความเป็นส่วนตัวของเราแล้ว
Licensed under cc by-sa 3.0 with attribution required.