Scala: 2次元配列を転置する方法
要素数 $M \times N$ の 2次元配列 $A$ に対して、$A_{ij} = B_{ji} \, (1 \le i \le M,\, 1 \le j \le N)$ となるような
$N \times M$ の2次元配列 $B=A^T$ を作成したい。
Scala の Array クラスには、そのものズバリ transpose というメソッドがある。
scala> val a = Array(Array(1,2,3),Array(4,5,6),Array(7,8,9),Array(10,11,12)) a: Array[Array[Int]] = Array(Array(1, 2, 3), Array(4, 5, 6), Array(7, 8, 9), Array(10, 11, 12)) scala> a.transpose res0: Array[Array[Int]] = Array(Array(1, 4, 7, 10), Array(2, 5, 8, 11), Array(3, 6, 9, 12))
これを利用すれば、文字列を区切って縦に結合するような処理もシンプルに記述できる。
scala> val s = """ | abc 123 | def 456 | """ s: String = " abc 123 def 456 " scala> s.split('\n').withFilter(!_.isEmpty).map(_.split("[ ]+")).transpose map (_.mkString) res1: Array[String] = Array(abcdef, 123456)
0 件のコメント:
コメントを投稿