Pythonのsort関数とitemgetterについて

Question

AtCoder Beginner Contest 128 B問題
のように1列目の要素を昇順にソートし同じだった場合2列目の要素を降順にソートするといったようなものを1行で書く方法を探しています。
最初に1列目をソートし値が同じものを新たなコンテナに入れそれをソートするという方法ではうまくいったのですが、できれば1行ですっきり書ける方法などあれば知りたいです。

例えば

from operator import itemgetter
a=[(5,3),(4,6),(2,6),(4,5),(9,2),(5,4)]
print(sorted(a,key=itemgetter(0,1)))
#[(2, 6), (4, 5), (4, 6), (5, 3), (5, 4), (9, 2)]

というようになることが分かりました。しかしどちらかにreverseを真にして一方は偽のままにするというようにすることができません。

sorted(a,key=itemgetter(0,1),reverse=itemgetter(False,True))

などとすると

print(sorted(a,key=itemgetter(0,1),reverse=itemgetter(False,True)))
TypeError: an integer is required (got type operator.itemgetter)

というエラーがでます。
ラムダ式を使ったものでも良いのでどなたか分かる方いましたらご教授のほどよろしくお願いいたします。

Answer 1

数字の場合は、こういう手↓が使えなくはありませんが、

>>> a=[(5,3),(4,6),(2,6),(4,5),(9,2),(5,4)]
>>> sorted(a, key=lambda x:(x[0], -x[1]))
[(2, 6), (4, 6), (4, 5), (5, 4), (5, 3), (9, 2)]

pythonのソートが安定ソートであることを利用して、2回に分けて書いた方が無難です。

>>> a=[(5,3),(4,6),(2,6),(4,5),(9,2),(5,4)]
>>> tmp = sorted(a, key=lambda x:x[1], reverse=True)
>>> result = sorted(a, key=lambda x:x[0])  # in-place処理のlist.sortの方がパフォーマンス上いいです
>>> result
[(2, 6), (4, 6), (4, 5), (5, 3), (5, 4), (9, 2)]

ソート HOW TO — Python 3.7.4rc1 ドキュメント

Comments

[Naru123]

ありがとうございます、in-place処理なども全く知らなかったです。
sortで使われているアルゴリズムが公式ドキュメントにも書かれていなかったのでクイックソートを使っていると勝手に思い込んでいましたが安定ソートなのですね、となると計算量はO(n*logn)なのでしょうか？

Answer 2

こんな感じでできます。

>>> sorted(a, key=lambda tpl: (tpl[0], -tpl[1]))
[(2, 6), (4, 6), (4, 5), (5, 4), (5, 3), (9, 2)]

Comments

[Naru123]

タプルを引数に持つ感じですかね、勉強になりました。
ありがとうございます。

[LouiS0616]

引数keyには一時比較用のオブジェクトを作るための関数オブジェクトを指定します。
operator.itemgetterもただの関数オブジェクトで、operator.itemgetter(n)(x) は x[n] と同じです。