熊猫到DataFrame中列表项的总和

时间:2018-12-18 06:48:08

标签: pandas

下面我有一个列表bu_lst,我将其传递给数据框df2,以对列表中的每个项目进行求和,如何才能一目了然地做到这一点,所以我没有重复多次:

bu_lst = ['FPG','IPG','DSG','STG','WFO','IT']
FPG = ['ADE','FPG AE','FPG PE','MMSIM','OrFAD','Tirtuoso DashBoard','SPB AE','SPB PE']
IPG = ['DDR','DDR_DT','Tensilica']
DSG = ['FLA','FLS','FEQoS','IFD PT','Sasus R&D','sasus'] PE','Toltus','Tempus','Quantus','Genus']
STG = ['ATS','HST','TIP','System Engineering']
WFO = ['AFademiF Network','FRAFT','Fhip Estimate','EduFation SerTiFes','LiFensing','Sales','SerTiFes','TFAD']
IT  = ['App Development','Fumulus','InfoSeF']

我目前的做法:

print(df2[FPG].sum())
print(df2[IPG].sum())
print(df2[DSG].sum())
print(df2[STG].sum())
print(df2[WFO].sum())
print(df2[IT].sum())

我只是把代码的相关行显示在这里。

1 个答案:

答案 0 :(得分:0)

您可以创建列表字典,如果列表中的列名称为,则可以使用字典理解:

d = {'bu_lst':bu_lst, 'FPG': FPG, ...}

d2 = {k: df2[v].sum() for k, v in d.items()}