web-dev-qa-db-ja.com

Scala、有限リストを無限に繰り返す

scalaでStreamクラスを使用して、指定されたリストを無限に繰り返したいと思います。

たとえば、リスト(1,2,3,4,5)を作成して、(1,2,3,4,5,1,2,3,4,5,1,2,3)を取得します。 ....)

テイク操作をラップできるように。私はこれが他の方法で実装できることを知っています、しかし私は何らかの理由でそれをこのようにしたいと思います、ただ私をユーモアを交えてください:)

つまり、あるリストから作成されたこの無限のサイクルで、take操作を使用でき、リストの最後に達すると循環するという考え方です。

特定のリストを単純に繰り返すストリームを作成するにはどうすればよいですか?

31
Felix

@Eastsunのものと非常に似ていますが、もう少し意図が明らかになります。 Scala 2.8でテスト済み。

scala> val l  = List(1, 2, 3)
l: List[Int] = List(1, 2, 3)

scala> Stream.continually(l.toStream).flatten.take(10).toList
res3: List[Int] = List(1, 2, 3, 1, 2, 3, 1, 2, 3, 1)

または、Scalazを使用します。

scala> import scalaz._
import scalaz._

scala> import Scalaz._
import Scalaz._

scala> val l = List(1, 2, 3)
l: List[Int] = List(1, 2, 3)

scala> l.toStream.repeat[Stream].join.take(10).toList
res7: List[Int] = List(1, 2, 3, 1, 2, 3, 1, 2, 3, 1)
36
retronym

別の方法は、入力の.toStreamをそれ自体と再帰的に連結することです。あれは、

scala> val xs: Stream[Int] = List(1, 2, 3).toStream #::: xs
xs: Stream[Int] = Stream(1, ?)

scala> xs.take(10).toList
res1: List[Int] = List(1, 2, 3, 1, 2, 3, 1, 2, 3, 1)
25
Volkan Yazıcı

scala 2.8でStream#flattenを使用する簡単な方法があります

Welcome to Scala version 2.8.0.r20542-b20100116020126 (Java HotSpot(TM) Client VM, Java 1.6.0_18).
Type in expressions to have them evaluated.
Type :help for more information.

scala> def cycle[T](seq: Seq[T]) = Stream.from(0).flatten(_ => seq)
cycle: [T](seq: Seq[T])scala.collection.immutable.Stream[T]

scala> cycle(1::2::3::Nil)
res0: scala.collection.immutable.Stream[Int] = Stream(1, ?)

scala> res0.take(10)
res1: scala.collection.immutable.Stream[Int] = Stream(1, ?)

scala> res0.take(10).toList
res2: List[Int] = List(1, 2, 3, 1, 2, 3, 1, 2, 3, 1)
7
Eastsun

lengthが効率的であると想定しない実装を次に示します。

def rep[A](seq: Seq[A]) = {
  def inner(proj: Seq[A]): Stream[A] = {
    if (proj.isEmpty)
      inner(seq)
    else
      Stream.cons(proj.first, inner(proj drop 1))
  }

  if (seq.isEmpty)
    Stream.empty
  else
    inner(seq)
}

これは、任意のSeqListまたはStreamを含む)に対して一定時間で実行する必要があり、各要素にデータを入力するために一定時間のオーバーヘッドのみを課します。また、無限のシーケンスでも機能します。したがって、無限のrepStreamを呼び出すことができ、結果のStreamは入力と同等になります。

5
Daniel Spiewak

優れたものから露骨に盗まれた Scala by Example 本、第12章、およびいくつかの変更を加えたもの:

def repeatedSeq(idx: Int, lst:Seq[Int]): Stream[Int] = Stream.cons(lst(idx), repeatedSeq((idx + 1)%lst.length, lst))

for(i <- repeatedSeq(1,List(1,1,2,3,5))) println(i)

これは、すべてのSeqタイプで機能します(もちろん、複数回読み取ることができない場合を除く)。 .length呼び出しが遅い場合は、効率的でない可能性があります。 Scala 2.7.7でテスト済み。

1
Justin W