python 多次元リストにおける最大値の抽出

Question

list = 
[('a', 'b', '97.4'), ('a', 'c', '69.3'), ('a', 'd', '100'), ('a', 'f', '95.9'), ('a', 'p', '68.9'), ('b', 'd', '64.7'), ('b', 'f', '191'), ('b', 'v', '213')]

このようなリストがあるときに、0番目の文字をキープしたまま2番目の最大値を抽出したいときはどうしたらいいでしょうか？

理想は

list_max = [('a', 'd', '100'),('b', 'v', '213')]

このように0番目の'a'の中で最も2番目の値が高いものをリスト化したいです。

再びエラーが発生してしまったので、編集追記させていただきます。

list = 
[('a', 'b', '100'), ('a', 'c', '69.3'), ('a', 'd', '100'), ('a', 'f', '95.9'), ('a', 'p', '68.9'), ('b', 'd', '64.7'), ('b', 'f', '191'), ('b', 'v', '213')]

>>> from collections import defaultdict
>>> d = defaultdict(list)
>>> for x in lst:
...     d[x[0]].append(x)
... 
>>> result = [max(x, key=lambda x:float(x[2])) for x in d.values()]
>>> result
[('a', 'b', '100'),('b', 'v', '213')]

大量のデータを扱っていましたら、このように

('a', 'b', '100'), ('a', 'd', '100')

('a', 'b', '100')が優先でmax値として拾われたために('a', 'd', '100')のdを
スルーしてしました。。。

この場合はどのように抽出していったいいでしょうか？
質問ばかりで申し訳ありません。宜しくお願い致します。

なぜ値が重複してしまったときにどちらも抽出したいかというと、
２つのリストから共通するものを取り出したいためです。

from collections import defaultdict


hoge = [('a', 'b', '1'), ('a', 'c', '2'), ('a', 'd', '3'),('d', 'j', '10'),('d', 'k', '10'),
               ('b', 'x', '1'), ('b', 'x', '2'), ('b', 'x', '3'), ('c', 'v', '1'), ('c', 'v', '2')]

hoge2 = [('b', 'a', '1'), ('c', 'a', '2'), ('d', 'a', '4'),
               ('d', 'k', '5'),
               ('mm', 'b', '1'), ('nn', 'b', '2'), ('x', 'b', '5'), ('kk', 'c', '6'), ('v', 'c', '3')]

print("最大値")
from collections import defaultdict
d = defaultdict(list)
for x in hoge:
    d[x[0]].append(x)

result = [max(x, key=lambda x:float(x[2])) for x in d.values()]
print(result)

print("最大値")
a = defaultdict(list)
for x in hoge2:
    a[x[0]].append(x)

result2 = [max(x, key=lambda x:float(x[2])) for x in a.values()]
print(result2)

print("最大値でdict作成")
hoge_dict = {(k1, k2):v for k1, k2, v in result}
hoge2_dict = {(k4, k3):v for k3, k4, v in result2}



print("作ったdictから共通するものをだす")
ok = {k:hoge[k] for k in hoge_dict.keys() & hoge2_dict.keys()}

print(ok)


コード

Answer 1

itertools.groupbyで。

import itertools
import operator


src = [
    ('a', 'b', '97.4'), ('a', 'c', '69.3'), ('a', 'd', '100'), ('a', 'f', '95.9'), ('a', 'p', '68.9'),
    ('b', 'd', '64.7'), ('b', 'f', '191'), ('b', 'v', '213')
]

dst = []
for _, it in itertools.groupby(src, key=operator.itemgetter(0)):
    m = max(it, key=lambda e: float(e[-1]))
    dst.append(m)

print(dst)

実行結果 Wandbox

[('a', 'd', '100'), ('b', 'v', '213')]

Answer 2

0番目の要素ごとにまとめた後、抽出するといいと思われます。

pandasとかでも書けそうですが、標準の機能の範囲でやるなら下のような感じです。

>>> lst = [('a', 'b', '97.4'), ('a', 'c', '69.3'), ('a', 'd', '100'), ('a', 'f', '95.9'), ('a', 'p', '68.9'), ('b', 'd', '64.7'), ('b', 'f', '191'), ('b', 'v', '213')]
>>> from collections import defaultdict
>>> d = defaultdict(list)
>>> for x in lst:
...     d[x[0]].append(x)
... 
>>> result = [max(x, key=lambda x:float(x[2])) for x in d.values()]
>>> result
[('b', 'v', '213'), ('a', 'd', '100')]

Comments

[22Go]

ありがとうございます！
defaultdictは知らなかったです。参考にさせていただきます！

[22Go]

すみません、リストの中身がこのように
('N_4434.1', 'N_273.1', '12')
なっているものを扱っていてエラーが出ました。
first argument must be callable or Noneと出たのですが、この場合どのように変更したらいいでしょうか？

[hayataka2049]

そのデータは特に問題なさそうに思われます。エラーメッセージを全文示してください。

[hayataka2049]

断定出来ませんが、listという変数名を使ってしまっているので、defaultdict(list)で失敗したような気がします。組み込み型・組み込み関数と被る変数名は避けるべきです。

[22Go]

仰る通りでした。変更したらできました。
ありがとうございます。