2016-11-21 6 views
1

私はタプルのリストを持っている:Pythonのグループと合計

Listoftuples=[ 
    (0.021892733407683305, 0.14887058717224647, 4.573173081530965, 0.04619366749021177, u'0102'), 
    (0.08416364174734663, 0.8500527816482009, 23.649983331004403, 0.0, u'0103'), 
    (0.02181070623592521, 0.15049387302788395, 1.2098398749067714, 1.6037412295275804, u'0102') 
    ] 

私はグループごとに(グループ=タプルの最後の値、例えばu'0102' )にしたい:

  • 集計最初に値
  • 最初の値の和で第二の値と除算を要約

試み:

import itertools 

Listoftuples=[ 
    (0.021892733407683305, 0.14887058717224647, 4.573173081530965, 0.04619366749021177, u'0102'), 
    (0.08416364174734663, 0.8500527816482009, 23.649983331004403, 0.0, u'0103'), 
    (0.02181070623592521, 0.15049387302788395, 1.2098398749067714, 1.6037412295275804, u'0102') 
    ] 

keyfunc=lambda t: (t[4]) 
Listoftuples.sort(key=keyfunc) 

for key,rows in itertools.groupby(Listoftuples, keyfunc): 
    sumOfFirstValue = sum(r[0] for r in rows) 
    sumOfSecondDividedBySumOfFirst= sum(r[1] for r in rows)/sumOfFirstValue 
    print key,sumOfFirstValue,sumOfSecondDividedBySumOfFirst 

結果:

0102 0.0437034396436 0.0 
0103 0.0841636417473 0.0 

最後でゼロ値。どうすれば修正できますか?

+0

'rows'がイテレータです。操作を実行する前にリストに変換してください。 –

答えて

5

よくある間違いは、groupbyから返されたrowsが具体的なリストであると考えることです。実際にはイテレータで、sumOfFirstValueの計算中に疲れました。この問題を回避するには、次のとおりです。

... 
for key,rows in itertools.groupby(Listoftuples, keyfunc): 
    rows = list(rows) 
    ... 
1

itertoolsを使用せずに簡単な解決策:

groups = set(item[4] for item in Listoftuples) 
for g in groups: 
    sum_first_val = sum([item[0] for item in Listoftuples if item[4] == g]) 
    sum_second_val = sum([item[1] for item in Listoftuples if item[4] == g]) 
    print g, sum_first_val, sum_second_val/sum_first_val