リストをKey関数、ラムダ式を使ってソートしたい

Question

student_tuples = [
    ('john', 'A', 15),
    ('jane', 'B', 12),
    ('dave', 'B', 10),
]
sorted(student_tuples, key=lambda student: student[2])   # sort by age
[('dave', 'B', 10), ('jane', 'B', 12), ('john', 'A', 15)]

このコードで、studentには('john', 'A', 15)、('jane', 'B', 12)、('dave', 'B', 10)がそれぞれ引き渡されて各タプルの3つ目の要素がkeyの引数として呼び出されているということでしょうか。

key=lambda student: student[2]の部分がどう動いているのかいまいちよく分からないのでご教授の程よろしくお願いいたします。

Answer 1

studentには('john', 'A', 15)、('jane', 'B', 12)、('dave', 'B', 10)がそれぞれ引き渡されて

ここまではご認識のとおりです。

各タプルの3つ目の要素がkeyの引数として呼び出されているということでしょうか。

いいえ。keyになっているのはあくまでもラムダ式です。
keyを利用して、『各タプルの3つ目の要素』を取り出しています。

key = lambda student: student[2]   # 関数オブジェクトの代入
for s in student_tuples:
    v = key(s)
    print(f'比較するときは{v}を使います。')

実行結果 Wandbox

比較するときは15を使います。
比較するときは12を使います。
比較するときは10を使います。

補足1

ラムダ式を変数に束縛することはあまり好まれません。
上記のコードは説明の為の悪い例で、通常はこのように書きます。

def key(student):
    return student[2]

...はい、これでも全く同じように取り扱えます。
慣れないうちはラムダ式を避け、関数定義を利用するのも一手でしょう。

補足2

リストやタプルなどの特定要素を取り出す機会は多いので、専用の関数が用意されています。

import operator

func = operator.itemgetter(2)
print(func(['a', 'b', 'c']))   # => c

次のうちどれを用いても、動作は全く同じです。

def func(student):
    return student[2]

sorted(student_tuples, key=func)

sorted(student_tuples, key=lambda student: student[2])

sorted(student_tuples, key=operator.itemgetter(2))

Comments

[aoi_]

大変丁寧にありがとうございます。
ラムダ式を使うのはあまり一般的ではなかったのですね。
他の方法まで教えていただいてとても助かりました。
誠にありがとうございました。

[LouiS0616]

> ラムダ式を使うのはあまり一般的ではなかったのですね。

いえ、そんなことないです。
慣れるまでは別の方法を使っても良いですけれど、余計な関数を作らない為にはラムダ式かoperator.itemgetterが適しています。