Can I use for-comprehenion / yield to create a map in Scala?

前端 未结 4 728
遥遥无期
遥遥无期 2021-01-01 15:19

Can I \"yield\" into a Map?

I\'ve tried

val rndTrans = for (s1 <- 0 to nStates;
                    s2 <- 0 to nStates
                            


        
4条回答
  •  青春惊慌失措
    2021-01-01 15:54

    An alternate solution in Scala 2.8:

    Welcome to Scala version 2.8.1.r23457-b20101106033551 (Java HotSpot(TM) Client VM, Java 1.6.0_22).
    Type in expressions to have them evaluated.
    Type :help for more information.
    
    scala> import scala.collection.breakOut            
    import scala.collection.breakOut
    
    scala> val list: List[(Int,Int)] = (for(i<-0 to 3;j<-0 to 2) yield(i->j))(breakOut)
    list: List[(Int, Int)] = List((0,0), (0,1), (0,2), (1,0), (1,1), (1,2), (2,0), (2,1), (2,2), (3,0), (3,1), (3,2))
    
    scala> val map: Map[Int,Int] = (for(i<-0 to 3;j<-0 to 2) yield(i->j))(breakOut)    
    map: Map[Int,Int] = Map((0,2), (1,2), (2,2), (3,2))
    
    scala> val set: Set[(Int,Int)] = (for(i<-0 to 3;j<-0 to 2) yield(i->j))(breakOut)
    set: Set[(Int, Int)] = Set((2,2), (3,2), (0,1), (1,2), (0,0), (2,0), (3,1), (0,2), (1,1), (2,1), (1,0), (3,0))
    
    scala> 
    

提交回复
热议问题