Scalaの多次元リスト

Question

###実現したいこと
以下に書いてあるコードをもう少し綺麗に書く良い方法はないでしょか？

###問題のコード

def hoge(list: list[List[String]]): List[List[List[List[String]]]] = {
  val a = for(i <- list) yield {
    val b = for(j <- i) yield {
      //処理
    }
    b 
  }
  a
}

Comments

[argius]

これはfor-yieldの戻り値として空の多次元リストが得られますが、それも意味があるということでよろしいでしょうか？
それとも単純に二重ループするだけで良いのでしょうか？

[Kuchitama]

hoge関数の戻り値型を書いていただけると、どのような結果が欲しいのか判断しやすくなると思います

Answer 1

あ、ベストアンサーになってしまいましたね...

編集していただきありがとうございます。ですが、これって綺麗に書く以前に現状で動作しませんよね？ コンパイルも通りませんし。実際に動かしたものを書いていただければ話が早いと思いますよ。

ご期待の処理かどうかは分かりませんが、とりあえず動くものを書いてみました。

  def hoge4(list: List[List[String]]): List[List[List[List[String]]]] = {
    list.map { x => x.map { y => List(List(y)) } }
  }

  val a = List(List("ABC", "DEF"), List("GHI"))
  println("--- hoge4 ---")
  println(hoge4(a))

結果

--- hoge4 ---
List(List(List(List(ABC)), List(List(DEF))), List(List(List(GHI))))

---

simple-teqさんがかなり詳しく回答されていらっしゃいますので、
私はおまけ的に回答を載せておきます。

hoge2はTraversable#foreachを使ったバージョン、
hoge3はforをまとめて書いたバージョンです。

いずれも戻り値は無し（Unit）です。

結果は異なりますが、お望みの方をお選び下さい。
戻り値が必要な場合は、simple-teqさんの回答をご覧ください。

def hoge2(list: List[List[String]]): Unit = {
list.foreach { i =>
  println(s"i: $i")
  i.foreach { j =>
    println(s"j: $j")
  }
}
}

def hoge3(list: List[List[String]]): Unit = {
for {
  i <- list
  j <- i
} {
  println(s"i: $i")
  println(s"j: $j")
}
}

val a = List(List("ABC", "DEF"), List("GHI"))
println("--- hoge2 ---")
hoge2(a)
println("--- hoge3 ---")
hoge3(a)

• 結果

--- hoge2 ---
i: List(ABC, DEF)
j: ABC
j: DEF
i: List(GHI)
j: GHI
--- hoge3 ---
i: List(ABC, DEF)
j: ABC
i: List(ABC, DEF)
j: DEF
i: List(GHI)
j: GHI

Comments

[退会済みユーザー]

ありがとうございます

Answer 2

試しに考えてみました。
hoge の引数は変更しております。

  def hoge(list: List[List[String]]) = {
    val a = for(i <- list) yield {
      val b = for(j <- i) yield {
        //処理
        j.toLowerCase()
      }
      b
    }
    a
  }

『引数のListの文字列を小文字に変換』を処理に記載しました。

実行すると下記の結果になります。

val hogeVal = hoge(List(List("A1","A2","A3"),List("B1","B2","B3") ))
println(hogeVal)
// List(List(a1, a2, a3), List(b1, b2, b3))

恐らく思った回答では無いのですが、こんな感じでfor文は入れ子に出来ます。

  def hoge_V2(list: List[List[String]]) = {
    for(
      i <- list;
      j <- i
    ) yield {
      j.toLowerCase()
    }
  }

この結果は、こんな感じです。
1次元のListになってしまいます（これはNG！？）。

val hogeVal_V2 = hoge_V2(List(List("A1","A2","A3"),List("B1","B2","B3") ))
println(hogeVal_V2)
// List(a1, a2, a3, b1, b2, b3)

こんな感じで map 関数でも実行できます。

def hoge_V3(list: List[List[String]]) = list map (i => i map(j => j.toLowerCase))

val hogeVal_V3 = hoge_V3(List(List("A1","A2","A3"),List("B1","B2","B3") ))
println(hogeVal_V3)
// List(List(a1, a2, a3), List(b1, b2, b3))

for , map , flatMap とかの関係が以下のGithubで紹介されています。
https://github.com/hatena/Hatena-Textbook/blob/master/foundation-of-programming-scala.md

読んでなんとなく、map、flatMapの動きはわかるのですが、私自身は身についている感じがまだしておりません。
これぐらいの簡単な例だとmapが一番短いですが、関数名とかも実用的な名称をつけたり、入れ子が深くなりすぎるとfor文（;でつなぐやり方）が見やすいのかもしれません。

Comments

[退会済みユーザー]

ありがとうございます