Python 並び替え 要素の種類によって昇順、降順を切り替える

Question

前提

Pythonでリストの並び替えをしたいです。

実現したいこと

list = [('1', 'f', 250), ('2', 'f', 400), ('3', 'f', 100), 
         ('4', 'm', 150), ('5', 'm', 700), ('6', 'm', 350)]

以上のリストを、
第2要素が'f'については、第３要素が昇順となるように並べ替え、
第2要素が'm'については、第３要素が降順となるように並べ替えたいのです。

list = [('3', 'f', 100), ('1', 'f', 250), ('2', 'f', 400), 
         ('5', 'm', 700), ('6', 'm', 350), ('4', 'm', 150)]

order_byやsortを使うと、全てに昇順、降順が適応されてしまいます。
リストを二つに分けてから、それぞれを並び替えて、リストをくっつけるなどはあまりしたくありません。
できるだけシンプルな処理で、並び替えができますでしょうか？

Answer 1

第一要素は無視ですかね？
sortやsortedでkeyに関数を指定して、ソートキーを作ります。

foo = [('1', 'f', 250), ('2', 'f', 400), ('3', 'f', 100),
         ('4', 'm', 150), ('5', 'm', 700), ('6', 'm', 350)]

foo.sort(key=lambda x:(x[1],x[2] if x[1]=='f' else -x[2]))

print(foo)

わかりにくければ、

def make_key(x):
    if x[1]=='f':
        return (x[1], x[2])
    else:
        return (x[1], -x[2])

foo.sort(key=make_key)

Comments

[reo037]

ifの内包表記でも可能なんですね！
関数での表記も記載していただき、ありがとうございます。

[otn]

内包表記ではないです。内包表記とは何なのか再確認しましょう。

Answer 2

lst = [('1', 'f', 250), ('2', 'f', 400), ('3', 'f', 100), 
       ('4', 'm', 150), ('5', 'm', 700), ('6', 'm', 350)]

lst_sorted = sorted(lst, key=lambda x: (x[1], x[2] * {'m': -1}.get(x[1], 1)))
print(lst_sorted)

Comments

[LouiS0616]

{'m': -1}.get(x[1], 1) の代わりに [-1, 1]['m' != x[1]] でも行けそうですね。
どっちが分かり易いかは... 微妙な線かもしれませんが。

[melian]

'f' や 'm' 以外にも増えるかもしれませんので。。。

[reo037]

＊の使い方や、getの使い方について勉強になりました。
ありがとうございます。