Python求和二维列表列重合

Question

我正在寻找最有效的方法来进行以下计算:

我有三个矩阵，像这样:

[[Brand1, operationCost], [Brand2, operationCost],...]

[[Brand1, maintenanceCost],[Brand2, maintenanceCost]...]

[[Brand1, replacementCost],[Brand2, replacementCost]...]

我需要计算每个品牌的总成本，运营+维护+更换。相同的标记可能不在所有矩阵中。并获得另一个像这样的矩阵:

[[Brand1, totalCost],[Brand2, totalCost]...]    

Answer 1

Numpy 应该可以解决您的问题:

例子:

import numpy as np
c = np.array([[1, 2, 3], [1, 2, 3]])
c.sum(0)
Out[5]: array([2, 4, 6])

如果你想让你的品牌与变量结合在一起，我会使用 pandas:

例子:

import pandas as pd
In[9]: df = pd.DataFrame({'brand1': [1, 2, 3], 'brand2': [1, 2, 3]})
In[10]: df
Out[10]: 
    brand1  brand2
0       1       1
1       2       2
2       3       3

In[11]: df.sum()
Out[11]: 
brand1    6
brand2    6