特定の数字を含む時の集計

Question

一番左側の数字が含む時に右側の数字を数字を集計したい。
取り敢えず、setで重複をなくして、inで確認を取りましたが、
ココから、どの様にすれば良いですか？

appendで別の配列に入れても、合計しか出ません。

#望む出力結果
```python
[[3,330],[4,95]]
```

#現状のコード


```python

r=[[3, 20, 70], [3, 40, 170], [3, 59, 90], [4, 5, 55], [4, 25, 40]] 

t=[i[0] for i in r]
t=list(set(t))

for i in t:
	for n in r:
#		print(i,n)
		if i in n:
			print(i,n[2])
"""
3 70
3 170
3 90
4 55
4 40
"""
```

#

過去の質問を参考にしましたが、引数が助長的になってしまいます。
改善する方法は有りますか？

#コード

```

r=[[3, 20, 70], [3, 40, 170], [3, 59, 90], [4, 5, 55], [4, 25, 40]] 

t=[i[0] for i in r]
t=list(set(t))

d = {}
for k,v,i in r:
    if k in d:
        d[k] += int(i)
    else:
        d[k] = int(i)

for k,v in d.items():
    print(k, v)

"""
3 330
4 95

"""
```

#
for文内のk,v,iなど省略したりできませんか？

Accepted Answer

tは最終的に使っていないようなので消したのと、不要なintを外したところだけ修正しましたが、質問文のコード
```python
r=[[3, 20, 70], [3, 40, 170], [3, 59, 90], [4, 5, 55], [4, 25, 40]] 

d = {}
for k,v,i in r:
    if k in d:
        d[k] += i
    else:
        d[k] = i

for k,v in d.items():
    print(k, v)
```
これが割とベターな気がするんですよね。シンプルでコードの姿も悪くないし。
更に単純化するとしたら、defaultdictを使ってキー有無の判定を省くことはできますが。
```python
from collections import defaultdict
r=[[3, 20, 70], [3, 40, 170], [3, 59, 90], [4, 5, 55], [4, 25, 40]] 

d = defaultdict(int)
for k,v,i in r:
    d[k] += i

for k,v in d.items():
    print(k, v)
```
これ以上単純化する方法は思いつきませんでした。ワンライナーで書くにしても、かえって黒魔術っぽくなっちゃうでしょう。そっちは他の方に譲ります（書く人がいれば）。
私が言いたいのは「上記コードで十分シンプルなのに、これ以上いじくり回す必要があるのか」ということです。

### 追記
他の方に譲る、と書いてしまったが・・・。
一行で書く別解。forをreduceで隠蔽しただけでロジックは同じ。実用上のメリットはないので、使わない方が良いです。
```python
from functools import reduce
from collections import defaultdict
r=[[3, 20, 70], [3, 40, 170], [3, 59, 90], [4, 5, 55], [4, 25, 40]] 
result_d = reduce(lambda d, e: (d.__setitem__(e[0], d[e[0]]+e[2]), d)[1], [defaultdict(int)]+r)
```
怪しい＿setitem＿をなくしたければ辞書をループの度に作って使い捨てれば良いですが、パフォーマンス上は不利です。そしてなおさら読めない代物になる（努力と工夫とひらめき次第でもうちょっと綺麗に書ける可能性はありますが）。

```python
from functools import reduce
r=[[3, 20, 70], [3, 40, 170], [3, 59, 90], [4, 5, 55], [4, 25, 40]] 
result_d = reduce(lambda d, e: dict((k, v+e[2]) if k == e[0] else (k, v) for k, v in d.items()) if e[0] in d else {k:v  for k, v in list(d.items())+[(e[0], e[2])]}, [{}]+r)
```
代入「文」は一行消費しちゃうので、これを使う処理は基本的に一行で綺麗には書けないことを覚えておくと良いでしょう。

Answer

1行で。
ほぼ、katoyさん。
インデックスが連続していないとソートしておく必要が。

```python
r=[[3, 20, 70], [3, 40, 170], [3, 59, 90], [4, 5, 55], [4, 25, 40]] 

from itertools import groupby

f = lambda r: {k:sum(x[-1] for x in v) for k, v in groupby(sorted(r, key=lambda x: x[0]), key=lambda x: x[0])}

a = f(r)
print(a)
```

Answer

```python
from itertools import groupby
from operator import itemgetter

r=[[3, 20, 70], [3, 40, 170], [3, 59, 90], [4, 5, 55], [4, 25, 40]]

d = []
for (k, g) in groupby(r, key=itemgetter(0)):
    items = [v for v in g]
    d.append([k, sum(item[2] for item in items)])

print(d)
```
実行結果
![イメージ説明](44253be88c8c95c9939db75e78e3cd98.png)

Answer

numpyのarrayに変換したほうが操作が簡単です。以下にtを作った後のコードを示します
```python
import numpy
result=[]
arr=numpy.array(r)
for i in t:
    tmp=numpy.sum(arr[arr[0]==i,2])
    result.append([i,tmp])
```