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

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)]