Aggiungere due elenchi di diverse lunghezze in python, a partire da destra

Question

voglio aggiungere due elenco di diversi inizio lunghezza dalla destra Ecco un esempio

[3, 0, 2, 1] 
[8, 7] 

Risultato atteso:

[3, 0, 10, 8] 

Questa lista rappresenta il coefficiente di polinomi

Ecco la mia implementazione

class Polynomial: 
    def __init__(self, coefficients): 
     self.coeffs = coefficients 

    def coeff(self, i): 
     return self.coeffs[-(i+1)] 

    def add(self, other): 
     p1 = len(self.coeffs) 
     p2 = len(other.coeffs) 
     diff = abs(p1 - p2) 
     if p1 > p2: 
      newV = [sum(i) for i in zip(self.coeffs, [0]*diff+other.coeffs)] 
     else: 
      newV = [sum(i) for i in zip([0]*diff+self.coeffs, other.coeffs)]     
     return Polynomial(newV) 

    def __add__(self, other): 
     return self.add(other).coeffs 

Questo lavoro va bene, voglio solo sapere in ogni caso di fare meglio, codice più pulito? Come pitone ha sempre sottolineato al codice pulito, voglio sapere c'è un modo per scrivere un codice più pulito e pitonioso?

Answer 1

>>> P = [3, 0, 2, 1] 
>>> Q = [8, 7] 
>>> from itertools import izip_longest 
>>> [x+y for x,y in izip_longest(reversed(P), reversed(Q), fillvalue=0)][::-1] 
[3, 0, 10, 8] 

Ovviamente, se si sceglie una convenzione in cui i coefficienti sono ordinate in senso opposto, si può semplicemente utilizzare

P = [1, 2, 0, 3] 
Q = [7, 8] 
[x+y for x,y in izip_longest(P, Q, fillvalue=0)] 

Answer 2

Credo che un semplice ciclo for è di gran lunga più semplice di una comprensione con zip_longest .. .

P = [3, 0, 2, 1] 
Q = [8, 7] 

A, B = sorted([P, Q], key=len) 

for i, x in enumerate(reversed(A), 1): 
    B[-i] += x 

#print(B) 

Se è necessario mantenere invariato P, copiarlo prima. Inoltre, se Q è molto più piccolo di P, questo sarà più efficace.

Answer 3

È possibile utilizzare numpy:

>>> import numpy as np 
>>> L1 = [3, 0, 2, 1] 
>>> L2 = [8, 7] 
>>> L2 = [0 for i in range(len(L1)-len(L2))] + L2 
>>> A1 = np.array(L1) 
>>> A2 = np.array(L2) 
>>> A1+A2 
array([ 3, 0, 10, 8]) 

Answer 4

a = [0, 1, 3, 4] 

b = [5, 6] 

iter_len = len(a)-(len(b) - 1) 

for j in range(len(a), 0, -1): 

     if j-iter_len < 0: 
      break 
     else: 
      a[j-1] = a[j-1] + b[j-iter_len] 
print a 

Answer 5

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

b = [1,2,5] 

n = a 

addStart = len(a) - len(b) 

count = 0 

for i in b : 

    n[addStart+count] = i + a[addStart+count] 
    count+=1 
print n