今回は無名関数lambda（ラムダ式）のより実践的な例として、辞書やリストをソートする方法を学びましょう。lambdaについてわからない場合はPython の無名関数 lambda（ラムダ式）の基本を参照ください。
辞書型オブジェクトをソート
値（バリュー）で並び替える場合
下のような辞書型オブジェクトを年齢順に並び替えて表示するにはどうすればよいでしょうか？
|
1 2 3 4 5 6 7 |
# 名前:年齢 の辞書型データ（dict） dict_staff = { 'suzuki': 51, 'tanaka': 25, 'yamada': 40, 'kimura': 33, } |
forとifを使って地道にソートする方法を真っ先に思いつくでしょうが、組み込み関数sorted()とlabmdaを使うとわずか１行でソートできてしまいます！
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
# 年齢昇順に並び替え sorted_staff = sorted(dict_staff.items(), key=lambda x: x[1]) # 表示して確認 for name, age in sorted_staff: print(name, age) ''' 出力内容 tanaka 25 kimura 33 yamada 40 suzuki 51 ''' |
sorted()関数は逆順にすることも簡単で、以下のようにreverse=Trueを指定すれば年齢降順に並び替わります。
|
1 |
sorted_staff = sorted(dict_staff.items(), key=lambda x: x[1], reverse=True) # reverse=True |
※組み込み関数sorted() について詳しく知りたい方は公式ドキュメントをチェックしてみましょう。
https://docs.python.jp/3/library/functions.html#sorted
キーで並び替える場合
辞書のキーと値（バリュー）が逆の場合はどうでしょう。つまり、年齢がキーで名前がバリューの辞書の場合です。
|
1 2 3 4 5 6 7 |
# 年齢:名前 の辞書オブジェクト dict_staff = { 51: 'suzuki', 25: 'tanaka', 40: 'yamada', 33: 'kimura', } |
辞書の値（バリュー）を基準にソートする場合には x[1] でしたが、キーを基準にソートする場合は x[0] にすればOKです。具体的には下記のようにすれば年齢順に表示することができます。
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
# 値を基準に並び替える場合は x[1] だったが、キーを規準にする場合は x[0] にする sorted_staff = sorted(dict_staff.items(), key=lambda x: x[0]) # 表示して確認 for age, name in sorted_staff: print(name, age) ''' 出力内容 tanaka 25 kimura 33 yamada 40 suzuki 51 ''' |
ただし、sorted()関数は引数keyを指定しない場合（デフォルトで）キーを基準にソートします。そのため、キーで並び替える場合にはlambdaが不要となります。※実際上は下のように記述するのがベターでしょう。
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
# sorted()はkeyを明示的に指定しない場合（デフォルトの場合）キーを基準に並び替えるので lambda は不要 sorted_staff = sorted(dict_staff.items()) # 表示して確認 for age, name in sorted_staff: print(name, age) ''' 出力内容 tanaka 25 kimura 33 yamada 40 suzuki 51 ''' |
※Github にこのページのソースコードをアップしています。
https://github.com/toksan/python3_study/blob/master/lambda/example_lambda_sort.py
タプルのリストをソート
公式ドキュメント（http://docs.python.jp/3.3/howto/sorting.html）で、タプルを要素に持つリストをソートするコードが紹介されています。（カタカナだらけでルー大柴さん状態で恐縮です）
公式ドキュメントのサンプル抜粋（一部コメントを補足）
|
1 2 3 4 5 6 7 8 9 10 11 12 |
# @see: http://docs.python.jp/3.3/howto/sorting.html student_tuples = [ ('john', 'A', 15), # name, grade ,age ('jane', 'B', 12), ('dave', 'B', 10), ] sorted_student_tuples = sorted(student_tuples, key=lambda student: student[2]) # sort by age # 補足： 0 name, 1 grade, 2 age のように0番目から数えるのでageを基準にソートする場合は student[2] になる # 確認 print(sorted_student_tuples) |
辞書のときと同様に組み込み関数sorted()とlabmdaを使うとわずか１行でソートできてしまいます！
Python の無名関数 lambda（ラムダ式）の基本では無名関数の有用性がいまいち伝わらなかったと思いますが、このように無名関数を使いこなすと非常に簡潔に記述できることがわかります。
※ map() や filter() とlabmdaの組み合わせもとても強力なので別の機会に紹介できればと思います。
コメントを残す