scala/for式入門

ここで学ぶこと

が、「ループ」が指せる範囲が他の言語よりも少し広い

Scalaのforの違い

意外となんでもループできる

forは式である

code:scala

for (i <- 0 until 9) {

// ^^^^^^^^^^^^^ ここをenumeratorと呼ぶ

// ここはbodyとか本体とか呼ぶ

println(i)

}

enumerator

body

enumeratorの次に来る箇所

code:javascript

// js

let str = '';

for (let i = 0; i < 9; i++) {

str = str + i;

} // for自体は値を返さない

式である = 値を返せる、ということ

code:scala

for (i <- 0 until 9) yield i

// => Seq(0, 1, 2, 3, 4, 5, 6, 7, 8)

これはまた次の節で説明します

forで値を返すには

code:scala

for (i <- 0 until 9) yield i * 2 // => Seq(0, 2, ... 14, 16)

多重for

code:scala

// この式は

for (i <- 0 to 10) {

for (j <- 0 to 10) {

println(s"$i * $j = ${i * j}")

}

}

// この式と等価

for (i <- 0 to 10; j <- 0 to 10) {

println(s"$i * $j = ${i * j}")

}

// enumeratorは()の中で;で区切る代わりに、{}の中で改行で区切ることもできる(こちらが一般的)

for {

i <- 0 to 10

j <- 0 to 10

} {

println(s"$i * $j = ${i * j}")

}

// 単一の式はそのまま書ける

for {

i <- 0 to 10

j <- 0 to 10

} println(s"$i * $j = ${i * j}")

code:scala

for {

i <- 'A' to 'Z'

j <- 1 to 5

} yield s"$i$j" // => Seq(A1, A2, A3, A4, A5, B1, B2, B3, B4, B5, C1, C2, C3, ...)

enumeratorの種類

code:scala

for {

i <- 1 to 10 // ジェネレータ

d = 2 // 定義

if i % d == 0 // フィルタ

} yield i // => Seq(2, 4, 6, 8, 10)

これらは基本的にどの順序で書いてもよい(書いた順に発動される)です

が、定義を先頭に書くことは今のところできません

これらを組み合わせて、柔軟なループと値の生成を実現できます

練習問題

code:scala

for {

n <- 1 to 50

} yield n match {

case n if n % 6 == 0 => "fizzbuzz"

case n if n % 2 == 0 => "fizz"

case n if n % 3 == 0 => "buzz"

case n => s"$n"

}

脱線 〜なんでもforで書けるよ編〜

ここまで読んだ時点で、たいていのループ処理は書ける

ここでおトクな知識を紹介します

map

code:scala

(0 to 5).map(i => i * 2) // => Seq(0, 2, 4, 6, 8, 10)

code:scala

for (i <- 0 to 5) yield i * 2

なるほど〜

filter

code:scala

(0 to 5).filter(i => i % 3 == 0).map(i => i * 2) //=> Seq(0, 6)

code:scala

for {

i <- 0 to 5

if i % 3 == 0

} yield i * 2 // => Seq(0, 6)

なるほど〜

flatMap

「全体に関数を適用し、その結果で新たにコレクションを作る」ことをインテリ用語で写像するとか射影すると言います

code:scala

List(List(1, 2), List(3), List(4, 5)).flatten // => List(1, 2, 3, 4, 5)

何が便利かというと:

コレクションの構造には口を出せない

それだけじゃ直感的に分からないので、具体的に見てみよう

code:scala

val xs = List(1, 2, 3, 4, 5)

// mapはコレクションの中身を操作するが、形を変えられない

xs.map(x => List(x, x)) // => List(List(1, 1), List(2, 2), ..., List(5, 5))

// flatMapは要素を潰して消したり、

xs.flatMap { x => x match {

case x if x % 2 == 0 => List()

case x => List(x)

}} // => List(1, 3, 5)

// 要素を増やしたりもできる

xs.flatMap(x => List(x, x + 1)) // => List(1, 2, 2, 3, 3, 4, 4, 5, 5, 6)

code:scala

val xs = List(1, 2, 3, 4, 5)

for {

x <- xs

x2 <- List(x, x + 1)

} yield x2 // => List(1, 2, 2, 3, 3, 4, 4, 5, 5, 6)

世界が隠した真実

な、なんだってー

真実の検証

code:scala

% scala-cli -Vprint:typer

scala> for {

| x <- List(1,2,3,4,5)

| x2 <- List(x, x + 1)

| } yield x2

package <empty> {

final lazy module val rs$line$3: rs$line$3 = new rs$line$3()

final module class rs$line$3() extends Object() { this: rs$line$3.type =>

{

{

def $anonfun(x2: Int): Int = x2

closure($anonfun)

}

)

closure($anonfun)

}

)

}

}