如何在第一个值相同的列表中对元组求和？

Question

我有一个股票和头寸列表作为元组.买入为正,卖出为负.例:

p = [('AAPL', 50), ('AAPL', -50), ('RY', 100), ('RY', -43)]

如何计算股票的头寸,以获得当前的持股？

result = [('AAPL', 0), ('RY', 57)]

Answer 1

这个怎么样？你可以阅读collections.defaultdict.

>>> from collections import defaultdict
>>> testDict = defaultdict(int)
>>> p = [('AAPL', 50), ('AAPL', -50), ('RY', 100), ('RY', -43)]
>>> for key, val in p:
        testDict[key] += val


>>> testDict.items()
[('AAPL', 0), ('RY', 57)]