lambda式 (無名関数)

lambda式 (無名関数)

lambda式(無名関数)

 コードの中で変数を使った簡単な定型処理をしたいけれど、関数として定義しておくほど大したものではなく (他のコードで再利用する予定もなく)、そのコードで使い捨てするような関数があれば便利です。Python の lambd式 (ラムダ式) がまさにそうした機能をもつ簡易関数です。lambd 式は基本的に

lambda 引数 : 式

という形で定義します (引数なしで定義することもできます)。lambda 式は名前をつけずに関数を宣言するので、無名関数 (匿名関数)とよばれることもあります。
 

lambda 式の基本的な使い方

 まずは lambda 式を使った簡単なコードを書いてみます。

# PYTHON_LAMBDA_01

# f(x) = |x|+2x
func = (lambda x : abs(x) + 2*x)

a = func(-1)
b = func(0)
c = func(1)

print("a =", a)
print("b =", b)
print("c =", c)
a = -1
b = 0
c = 3

 このサンプルコードでは lambda 式を使って |x| + 2x という関数を定義しています。
 同じ処理を通常の def 構文を使って書くと次のようになります。

# PYTHON_LAMBDA_02

# f(x) = |x| + 2x
def func(x):
    return abs(x) + 2*x

a = func(-1)
b = func(0)
c = func(1)

print("a =", a)
print("b =", b)
print("c =", c)
a = -1
b = 0
c = 3

 このように、def文を使うと改行とインデントが入るので、簡単な関数を定義する場合でも最低 2 行のコードが必要になりますが、lambda 式を使えば 1 行で記述することができます。
 

lambda 式と条件分枝

 上の例では、どちらのスタイルで書いてもたった 1 行しか変わらないので lambda 式の有難味がよくわからないかもしれません。もう少し複雑なコードで比較してみましょう。数学の絶対値の問題でよく出てくるような「変数 x が負のときは |x|, 正のときは x を返す関数」です。まずは def文を使ったコードです。

# PYTHON_LAMBDA_03

def func(x):
    if x < 0:
        return abs(x)
    else:
        return x

a = func(-5)
b = func(5)

print("a =", a)
print("b =", b)
a = 5
b = 5

 def文の中に if を使った条件分枝が入るので、関数を定義するために 5 行のコードが必要です。lambda 式を使うと 1 行で済みます:

# PYTHON_LAMBDA_04

func = (lambda x : abs(x) if x < 0 else x)

a = func(-5)
b = func(5)

print("a =", a)
print("b =", b)
a = 5
b = 5

 このように、lambda 式の中に条件分枝を含む処理を書き込むこともできます。やろうと思えば、もっと複雑な処理を 1 行の中に押し込むこともできますが、そうすると可読性が落ちてしまうので、ほどほどにしておいたほうがいいようです。
 

lambda式によるソート

 リストの並び替えの記事でも扱いましたが、sort()メソッドの引数に lambda 式を渡してリストの要素を並び替えることができます。たとえば、整数のペア (a, b) を要素にもつリストがあって、a の値について昇順(小さい順)に並び替えたいときには、次のようなコードを書きます。

# PYTHON_LAMBDA_05

# タプルを要素にもつリストを定義
my_list = [(7,5), (2, 4), (9, 0), (8, 8)]

# インデックス0の要素をキーとして昇順に並び替え
my_list.sort(key = lambda x : x[0])

print(my_list)
[(2, 4), (7, 5), (8, 8), (9, 0)]