Haskellの文法に慣れて来た方を対象に、ラムダ式や高階関数を使って関数を取り回す方法を説明します。カリー化や部分適用も取り上げます。いわゆる関数型言語らしい機能です。
シリーズの記事です。
- Haskell 超入門
- Haskell 代数的データ型 超入門
- Haskell アクション 超入門
- Haskell ラムダ 超入門 ← この記事
- Haskell アクションとラムダ 超入門
- Haskell IOモナド 超入門
- Haskell リストモナド 超入門
- Haskell Maybeモナド 超入門
- Haskell 状態系モナド 超入門
- Haskell モナド変換子 超入門
- Haskell 例外処理 超入門
- Haskell 構文解析 超入門
- 【予定】Haskell 継続モナド 超入門
- 【予定】Haskell 型クラス 超入門
- 【予定】Haskell モナドとゆかいな仲間たち
- 【予定】Haskell Freeモナド 超入門
- 【予定】Haskell Operationalモナド 超入門
- 【予定】Haskell Effモナド 超入門
- 【予定】Haskell アロー 超入門
練習の解答例は別記事に掲載します。
この記事には@shigemk2さんによるScala版があります。
この記事はとある方への誕生日プレゼントとして捧げます。
ラムダ式
今まで取り上げて来た文法では、関数の引数を左辺で定義していました。
inc x = x + 1
main = do
print $ inc 5
6
引数を右辺で定義する文法があります。
inc = \x -> x + 1
main = do
print $ inc 5
6
この右辺をラムダ式と呼びます。\（バックスラッシュ）をギリシア文字の λ（ラムダ）に見立てています。
※ Windowsで\は円記号で表示されることがありますが、本来のASCIIではバックスラッシュです。
練習
【問1】次に示す関数factをラムダ式とcase～ofで書き換えてください。
fact 0 = 1
fact n | n > 0 = n * fact (n - 1)
main = do
print $ fact 5
⇒ 解答例
型注釈
型注釈とラムダ式を並べると、型注釈の書式がラムダ式と共通していることが分かります。
inc :: Int -> Int
inc = \x -> x + 1
main = do
print $ inc 5
6
練習
【問2】次に示す関数addをラムダ式で書き換えてください。
add :: Int -> Int -> Int
add x y = x + y
main = do
print $ add 2 3
⇒ 解答例
無名関数
ラムダ式は名前のない関数（無名関数）で、それを変数に束縛していると捉えることができます。
a = 1
b = \x -> x + 1
main = do
print $ b a
2
ラムダ式を束縛しないで使うこともできます。
main = do
print $ (\x -> x + 1) 1
2
一度しか使わない関数にわざわざ名前を付けるのが面倒なとき、ラムダ式は便利です。
複数の引数
次の3つはすべて同じ型Int -> Int -> Intです。
mul x y = x * ymul = \x y -> x * ymul = \x -> \y -> x * y
1と2は3の糖衣構文です。
練習
【問3】次に示す関数addを定義せずに、呼び出し側で無名関数にインライン展開してください。
add x y = x + y
main = do
print $ add 2 3
⇒ 解答例
高階関数
引数として関数を受け取ったり、戻り値として関数を返したりする関数を高階関数と呼びます。
引数
引数として関数を受け取る高階関数の例です。
f g = g 2 3
add x y = x + y
mul = \x y -> x * y
main = do
print $ f add
print $ f mul
5
6
上のmulはラムダ式が変数に束縛され、その変数をfに渡しています。変数を経由せずにラムダ式を直接渡すこともできます。
f g = g 2 3
main = do
print $ f $ \x y -> x + y
print $ f $ \x y -> x * y
5
6
戻り値
戻り値として関数を返す高階関数の例です。
add x = \y -> x + y
main = do
let add2 = add 2
print $ add2 3
print $ (add 2) 3
print $ add 2 3
5
5
5
上のaddはラムダ式\y -> x + yで表される無名関数を返す高階関数です。高階関数から戻された関数に後続の引数を渡すことで連続して呼び出すことができます。
練習
【問4】次に示す関数fとaddを定義せずに、呼び出し側で無名関数にインライン展開してください。
f g = g 1 2
add x y = x + y
main = do
print $ f add
⇒ 解答例
【問5】次に示す関数addを定義せずに、呼び出し側で無名関数にインライン展開してください。
add x = \y -> x + y
main = do
print $ add 1 2
⇒ 解答例
カリー化
複数の引数を取る関数に対して、引数を後ろから1つずつ右辺に移動させてみます。
add x y = x + y
add' x = \y -> x + y
add'' = \x -> \y -> x + y
main = do
print $ add 2 3
print $ add' 2 3
print $ add'' 2 3
5
5
5
どれも同じように振る舞うため、定義は等価だと見なせます。このように引数を1つずつ分割して関数をネストさせることをカリー化と呼びます。Haskellでは複数の引数を取る関数は自動的にカリー化されます。
部分適用
引数が足りない場合、後で付け足せば呼び出しを完成させることができます。このようなことが可能になるのも関数がカリー化されているためです。
add x y = x + y
main = do
let add2 = add 2
print $ add2 3
print $ (add 2) 3
print $ add 2 3
5
5
5
上のadd2のように一部の引数を固定化して新しい関数を作り出すことを部分適用と呼びます。
注意点
カリー化と部分適用は混同されることがあります。具体的には、部分適用を指してカリー化と呼ばれることがありますが、これは誤用です。
改めて定義を確認します。
- カリー化とは：関数を引数1つずつに分割してネストさせること
- 部分適用とは：一部の引数を固定化して新しい関数を作り出すこと
詳細は次の記事にまとめました。
- @7shi: カリー化と部分適用（JavaScriptとHaskell） - Qiita 2014.10.15
練習
【問6】次に示す関数combineを、引数1つずつに分割してネストさせたラムダ式で書き換えてください。
combine a b c = a:b:[c]
main = do
let a = combine 1
b = a 2
c = b 3
print c
print $ combine 'a' 'b' 'c'
⇒ 解答例
【問7】次のコードから関数doubleを除去してください。ラムダ式は使わないでください。
f xs g = [g x | x <- xs]
double x = 2 * x
main = do
print $ f [1..5] double
ヒント: 演算子の関数化(*)
⇒ 解答例
演算子
演算子を関数化すれば高階関数に渡すことができます。
f g = g 2 3
main = do
print $ f (+)
print $ f (*)
5
6
部分適用
演算子を関数化すれば部分適用できます。（問7で出題）
f g = g 5
main = do
print $ f $ (-) 2
-3
セクション
中置演算子のまま片方のオペランド（被演算子）を省略した不完全な式は部分適用として扱われます。これをセクションと呼びます。セクションは括弧で囲む必要があります。
ラムダ式とセクションを対比します。(- 2)はマイナス2という数値として解釈されるため、2種類の回避方法を示しています。
f g = g 5
main = do
print [f $ \x -> 2 + x, f (2 +)]
print [f $ \x -> x + 2, f (+ 2)]
print [f $ \x -> 2 - x, f (2 -)]
print [f $ \x -> x - 2, f (+(-2)), f $ subtract 2]
[7,7]
[7,7]
[-3,-3]
[3,3,3]
関数を演算子化すればセクションとして扱えます。
f g = g 5
main = do
print $ f (2 `div`)
print $ f (`div` 2)
0
2
セクションは次の記事を参考にしました。
- @jutememo: Haskell のセクションと中置記法 | すぐに忘れる脳みそのためのメモ 2008.6.24
- @kk_ataka: すごいHaskellたのしく学ぼうでHaskellことはじめ4 - kk_Atakaの日記 2013.2.20
練習
【問8】次のコードからラムダ式を排除してください。
f1 g = g 1
f2 g = g 2 3
main = do
print $ f1 $ \x -> x - 3
print $ f1 $ \x -> 3 - x
print $ f2 $ \x y -> x + y
⇒ 解答例
色々な関数
Prelude（標準ライブラリ）で定義されている高階関数をいくつか紹介します。
map
リストの要素すべてに同じ処理を施した別のリストを作成します。
同じことができるリスト内包表記と対比します。
main = do
print $ map (* 2) [1..5]
print [x * 2 | x <- [1..5]]
[2,4,6,8,10]
[2,4,6,8,10]
リスト内包表記との使い分けは特に基準はありませんが、高階関数の扱いに慣れればリスト内包表記が冗長に感じるかもしれません。
filter
リストから要素を取り出す際に条件を指定できます。
同じことができるリスト内包表記と対比します。
main = do
print $ filter (< 5) [1..9]
print [x | x <- [1..9], x < 5]
[1,2,3,4]
[1,2,3,4]
flip
2引数関数で引数の順序を反転します。
第2引数への部分適用をセクションやラッパーと対比します。
src = [1..5]
test1 = flip map src
test2 = (`map` src)
test3 f = map f src
main = do
print $ test1 (* 2)
print $ test2 (* 2)
print $ test3 (* 2)
[2,4,6,8,10]
[2,4,6,8,10]
[2,4,6,8,10]
第1引数がラムダ式で記述すると長くなる時、先に第2引数を記述するために使うと便利です。Yコンビネータで実例を出します。
foldl
リストの要素を左から1つずつ処理しながら集計します。関数名末尾のlはLeft（左）の意味です。
foldl :: 集計関数 -> 初期値 -> リスト -> 集計結果
foldlでsum相当の処理をしてみました。
main = do
print $ sum [1..100]
print $ foldl (+) 0 [1..100]
5050
5050
foldlは手続型言語のループを関数化したものだと見なせます。次のJavaScriptコードと比較してみてください。
var sum = 0;
for (var i = 1; i <= 100; ++i) {
sum += i;
}
console.log(sum);
5050
集計の初期値として0を用意して、ループでiを次々に足していきます。そこから処理（足し算）と初期値を取り出して関数化したのがfoldlだと見なすわけです。
foldr
リストの要素を右から1つずつ処理しながら集計します。関数名末尾のrはRight（右）の意味です。
main = do
print $ foldr (-) 0 [1..5]
3
再帰でリストの全要素を処理する際に、再帰から返って来た値を使って関数の戻り値を計算すると、戻り値が確定するのは再帰の復路です。この手の再帰を関数化したものだと見なせます。次のコードと比較してみてください。
test [] = 0
test (x:xs) = x - test xs
main = do
print $ test [1..5]
3
再帰の折り返し値として0を用意して、復路でxから次々に引いていきます。そこから処理（引き算）と折り返し値（最初に作用させる値）を取り出して関数化したのがfoldrだと見なすわけです。
-
1 - (2 - (3 - (4 - (5 - 0))))→1 - 2 + 3 - 4 + 5→3
※ 復路で実際の計算が始まるため、計算の順はリストの右からとなります。これがRightの意味です。
foldrは次の記事を参考にしました。
練習
【問9】map, filter, flip, foldl, foldrを再帰で再実装してください。関数名には'を付けてください。
具体的には以下のコードが動くようにしてください。
main = do
print $ map' (* 2) [1..5]
print $ filter' (< 5) [1..9]
print $ flip' map' [1..5] (* 2)
print $ foldl' (+) 0 [1..100]
print $ foldl' (-) 0 [1..5]
print $ foldr' (-) 0 [1..5]
[2,4,6,8,10]
[1,2,3,4]
[2,4,6,8,10]
5050
-15
3
⇒ 解答例
【問10】foldlでreverseとmaximumとminimumを再実装してください。関数名には'を付けてください。
具体的には以下のコードが動くようにしてください。
main = do
let src = [-5..5]
print $ reverse' src
print $ maximum' src
print $ minimum' src
[5,4,3,2,1,0,-1,-2,-3,-4,-5]
5
-5
⇒ 解答例
【問11】次に示す関数qsortをfilterで書き替えてください。
qsort [] = []
qsort (n:xs) = qsort lt ++ [n] ++ qsort gteq
where
lt = [x | x <- xs, x < n]
gteq = [x | x <- xs, x >= n]
main = do
print $ qsort [4, 6, 9, 8, 3, 5, 1, 7, 2]
⇒ 解答例
【問12】次に示す関数bswapをfoldrで書き替えてください。
※ バブルソートの本体はfoldrでは実装できないため省きました。
bswap [x] = [x]
bswap (x:xs)
| x > y = y:x:ys
| otherwise = x:y:ys
where
(y:ys) = bswap xs
main = do
print $ bswap [4, 3, 1, 5, 2]
[1,4,3,2,5]
⇒ 解答例
この問題は次の記事を参考にしました。
- @kazu_yamamoto: リストの畳み込みと展開 - あどけない話 2011.9.13
不動点コンビネータ
自己参照のできない無名のラムダ式で再帰を実現するテクニックとして、不動点コンビネータを利用する方法があります。あまり使う機会はないかもしれませんが、たまに見掛けるので知識として知っておいても損はないでしょう。
Yコンビネータ
Yコンビネータ（不動点コンビネータの一種）と呼ばれる補助関数を定義します。
y f = f (y f)
Yコンビネータにラムダ式を渡すと、ラムダ式の第1引数にYコンビネータに包まれた自分自身が渡されます。これを使うことで再帰ができます。
sumをインラインで実装した例です。ラムダ式の第1引数を関数名に見立てています。
y f = f (y f)
main = do
print $ flip y [1..100] $
\sum xs -> case xs of
[] -> 0
(x:xs') -> x + sum xs'
5050
Yコンビネータは次の記事を参考にしました。
- @kazu_yamamoto: Haskell で Y コンビネータ - あどけない話 2010.5.19
- @kazu_yamamoto: Yコンビネータのまとめ - あどけない話 2010.6.1
練習
【問13】Yコンビネータを使って10番目のフィボナッチ数を計算してください。
⇒ 解答例
関数合成
関数を連続して呼ぶのに対して、.で関数を合成しても同じことができます。
f x = x + 1
g x = x * 2
main = do
print $ f (g 1)
print $ (f . g) 1
3
3
関数を評価する順番はどちらも g → f です。
関数を使うだけだとあまり違いは感じませんが、高階関数に渡すときに関数合成は便利です。
f x = x + 1
g x = x * 2
h f = f 1
main = do
print $ h $ f . g
print $ h $ \x -> f $ g x
3
3
2引数
gの引数が増えるといきなり難しくなります。
f x = x * 2
g x y = x + y
main = do
print $ f $ g 1 2 -- (1 + 2) * 2
print $ ((f .) . g) 1 2 -- 関数合成
6
6
初期の段階であまり追及するようなものではないため紹介程度に留めておきますが、興味がある方は次の記事を参照してください。
- @7shi: Haskell - 関数合成を機械的に扱う試み - Qiita 2015.2.9
ポイントフリースタイル
部分適用を利用すれば、別の関数に渡すだけの引数を省略できます。これをポイントフリースタイルと呼びます。
簡単な例を示します。
f1 x y = x - y
f2 x y = (-) x y
f3 x = (-) x
f4 = (-)
g1 x = f4 2 x
g2 = f4 2
main = do
print $ g2 5
-3
関数合成と組み合わせれば色々なパターンで引数を排除することができます。ただあまり突き詰めると、すぐには読めないコードになるような印象があります。
もっと恐ろしいものの片鱗を味わいたい方は、次の記事を参照してみてください。
- @melponn: ポイントフリースタイル入門 - melpon日記 2011.10.31