Déballer itérables en utilisant la carte

Question

En supposant que j'ai deux iterables de chiffres de la même longueur

weights = range(0, 10) 
values = range(0, 100, 10) 

je dois compter la somme pondérée. Je sais que cela peut être fait avec la compréhension de la liste

weighted_sum = sum(weight * value for weight, value in zip(weights, values)) 

Je me demande si cela peut être fait en utilisant map et operator.mul comme

import operator 

weighted_sum = sum(map(operator.mul, zip(weights, values))) 

mais cela donne une erreur

Traceback (most recent call last): 
    File "<input>", line 3, in <module> 
TypeError: op_mul expected 2 arguments, got 1 

donc mon question: existe-t-il un moyen de faire passer les tuples déballés à l'aide de map?

Answer 1

map n'a pas besoin du zip, il suffit d'utiliser

weighted_sum = sum(map(operator.mul, weights, values)) 

De map's Documentation

Si d'autres arguments itératifs sont passés, la fonction doit prendre autant d'arguments et être appliquée aux éléments de tous les itérables en parallèle.

également mentionné dans la documentation de map est qu'au lieu que vous pouvez utiliser itertools.starmap au lieu de map pour déjà zip entrée ped.

---

Comme Rahul hinted at, en utilisant numpy est toujours une bonne idée lorsqu'il s'agit de valeurs numériques, en fait quelque chose comme

import numpy as np 

np.asarray(weights) * values 

devrait déjà faire l'affaire (mais contrairement à map ce nécessite les deux tableaux être de même longueur, tandis que map mapperait la longueur la plus courte).

Answer 2

Essayez ceci:

>>> import operator 
>>> 
>>> weights = range(0, 10) 
>>> values = range(0, 100, 10) 
>>> sum(map(lambda i:operator.mul(*i), zip(weights, values))) 
2850 

Ou

>>> sum(map(operator.mul, weights, values)) 
2850 

Answer 3

Vous pouvez également essayer avec numpy,

In [45]: import numpy as np 

In [46]: sum(map(np.multiply,weights,values)) 
Out[46]: 2850 

Selon Tobias Suggestion de Kienzler,

In [52]: np.sum(np.array(weights) * values) 
Out[52]: 2850