Mam następujące dwie listy:
first = [1,2,3,4,5]
second = [6,7,8,9,10]
Teraz chcę dodać elementy z obu tych list do nowej listy.
wyjście powinno być
third = [7,9,11,13,15]
Mam następujące dwie listy:
first = [1,2,3,4,5]
second = [6,7,8,9,10]
Teraz chcę dodać elementy z obu tych list do nowej listy.
wyjście powinno być
third = [7,9,11,13,15]
Odpowiedzi:
zip
Funkcja jest przydatna tutaj, używany z listowego.
[x + y for x, y in zip(first, second)]
Jeśli masz listę list (zamiast tylko dwóch):
lists_of_lists = [[1, 2, 3], [4, 5, 6]]
[sum(x) for x in zip(*lists_of_lists)]
# -> [5, 7, 9]
first
ma długość 10, a second
długość 6, wynik spakowania iteracji będzie miał długość 6.
>>> lists_of_lists = [[1, 2, 3], [4, 5, 6]]
>>> [sum(x) for x in zip(*lists_of_lists)]
[5, 7, 9]
import operator
list(map(operator.add, first,second))
Domyślnym zachowaniem w numpy jest dodawanie komponentów
import numpy as np
np.add(first, second)
które wyjścia
array([7,9,11,13,15])
Zakładając, że obie listy a
i b
mają taką samą długość, nie trzeba zip, numpy lub cokolwiek innego.
Python 2.x i 3.x:
[a[i]+b[i] for i in range(len(a))]
Dotyczy to dowolnej liczby list:
[sum(sublist) for sublist in itertools.izip(*myListOfLists)]
W twoim przypadku myListOfLists
byłoby[first, second]
izip.from_iterable
?
chain
. Zaktualizowano
Wypróbuj poniższy kod:
first = [1, 2, 3, 4]
second = [2, 3, 4, 5]
third = map(sum, zip(first, second))
map(sum, zip(first, second, third, fourth, ...))
Najłatwiejszym i szybkim sposobem na to jest:
three = [sum(i) for i in zip(first,second)] # [7,9,11,13,15]
Alternatywnie możesz użyć numpy sum:
from numpy import sum
three = sum([first,second], axis=0) # array([7,9,11,13,15])
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]
lambda x, y: x + y
może być po prostu sum
.
Jeśli masz nieznaną liczbę list o tej samej długości, możesz użyć poniższej funkcji.
Tutaj * args akceptuje zmienną liczbę argumentów listy (ale sumuje tylko taką samą liczbę elementów w każdym). Znak * jest ponownie używany do rozpakowywania elementów na każdej z list.
def sum_lists(*args):
return list(map(sum, zip(*args)))
a = [1,2,3]
b = [1,2,3]
sum_lists(a,b)
Wynik:
[2, 4, 6]
Lub z 3 listami
sum_lists([5,5,5,5,5], [10,10,10,10,10], [4,4,4,4,4])
Wynik:
[19, 19, 19, 19, 19]
rozwiązanie jednowarstwowe
list(map(lambda x,y: x+y, a,b))
Moja odpowiedź jest powtarzana z Thiru, który odpowiedział na nią 17 marca o 9:25.
Było prościej i szybciej, oto jego rozwiązania:
Najłatwiejszym i szybkim sposobem na to jest:
three = [sum(i) for i in zip(first,second)] # [7,9,11,13,15]
Alternatywnie możesz użyć numpy sum:
from numpy import sum three = sum([first,second], axis=0) # array([7,9,11,13,15])
Potrzebujesz numpy!
tablica numpy może wykonywać pewne operacje, takie jak wektoryimport 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]
Możesz użyć zip()
, który „przepleci” dwie tablice razem, a następnie map()
zastosuje funkcję do każdego elementu w iterowalnej:
>>> 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]
Oto inny sposób, aby to zrobić. Korzystamy z wewnętrznej funkcji __add__ w Pythonie:
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)
Wynik
[11, 22, 33, 44, 55]
Jeśli chcesz dodać również resztę wartości na listach, możesz tego użyć (to działa w Pythonie 3.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))
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]
Oto inny sposób na zrobienie tego, który dla mnie działa dobrze.
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("")
j = min(len(l1), len(l2))
l3 = [l1[i]+l2[i] for i in range(j)]
Być może najprostsze podejście:
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)
Co jeśli masz listę o różnej długości, możesz spróbować czegoś takiego (używając 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]
Możesz użyć tej metody, ale zadziała ona tylko wtedy, gdy obie listy mają ten sam rozmiar:
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