两组 Python 计算

Question

我有以下两个列表：

          level = ["a","b","a","c"]
          Number = [1,2,1,2]

我只是想知道是否可以使用 Python 计算列表级别中每个元素的总和。所以输出将是两个新列表 x 和 y：

          x = ["a", "b","c"]
          y = [2,2,2]  (e.g: the total Number for "a" is 1+1=2, etc)

Answer 1

一种方法是使用defaultdict:

from collections import defaultdict
d = defaultdict(int)
for k,v in zip(level, frequency):
    d[k] += v

另一种方法是使用Counter：

from collections import Counter
import itertools
d = Counter(itertools.chain.from_iterable(map(itertools.repeat, level, frequency)))

在这两种方式中，您都可以将其解压缩为 2 个列表，如下所示：

x, y = zip(*d.items())

Answer 2

您必须为它们创建键映射。

在下面的代码中，我创建了带有映射的字典。并检查，如果您的level 不存在于dict 中，则将您的frequency 与0 相加，否则与最后计算的frequency 相加。

>>> level = ["a","b","a","c"]
>>> frequency = [1,2,1,2]
>>> my_dict = {}
>>> for (l, f) in zip(level, frequency):
...     my_dict[l] = my_dict.get(l, 0) + f
... 
>>> print my_dict
{'a': 2, 'c': 2, 'b': 2}
>>> print my_dict.values()
[2, 2, 2]
>>> print my_dict.keys()
['a', 'c', 'b']

Answer 3

如果您不关心x 中的顺序，您可以使用defaultdict，尽管我确信有更实用的方法来做到这一点

from collections import defaultdict
defdic = defaultdict(int)
for letter, frequency in zip(level, frequency):
    defdic[letter] += frequency

x,y = list(defdic.keys()), list(defdic.values())

# or better yet as proposed by Bharel
x, y = zip(*defdic.items())

输出：

>>> x 
['a', 'b', 'c']
>>> y 
[2, 2, 2]