Scala使用键减少Case类

时间:2016-02-25 14:16:31

标签: scala dictionary reduce

以下是我有2个案例类的情景 -

case class Pack(name: String, age: Int, dob: String)

object Pack{
  def getPack(name:String, age: Int, dob:String): Pack = {
    Pack(name,age,dob)
  }
}

case class NewPack(name: String, pack: (List[Pack]))

object NewPack{
  def getNewPackList(data: List[Pack]): List[NewPack] = {
    val newData = for(x <- data )yield(x.name,List(x))
    val newPackData = for(x <- newData)yield(NewPack(x._1,x._2))
    newPackData
  }
}


val someData = List(Pack.getPack("x",12,"day1"), Pack.getPack("y",23,"day2"),Pack.getPack("x",34,"day3") )

val somePackData = NewPack.getNewPackList(someData)

someData和somePackData的值是这样的 -

someData: List[Pack] = List(Pack(x,12,day1), Pack(y,23,day2), Pack(x,34,day3))

somePackData: List[NewPack] = List(NewPack(x,List(Pack(x,12,day1))), NewPack(y,List(Pack(y,23,day2))), NewPack(x,List(Pack(x,34,day3))))

现在我想以下面提到的格式获取最终数据。任何人都可以建议我更好的方法。

finalData: List[NewPack] = List(NewPack(x,List(Pack(x,12,day1),Pack(x,34,day3)),NewPack(y,List(Pack(y,23,day2))))

1 个答案:

答案 0 :(得分:3)

这是一个小小的简化。

case class Pack(name: String, age: Int, dob: String)

case class NewPack(name: String, pack: List[Pack])

object NewPack{
  def getNewPackList(data: List[Pack]): List[NewPack] = 
    data.map{case pack@Pack(name, _, _) => NewPack(name, List(pack))}
}

请注意,一般情况下,您不需要为案例类提供额外的工厂方法。如果您想稍后提出一些棘手的问题,可以随时在随播对象中定义自定义apply

此外,如果您打算按名称对包进行分组,则可以轻松定义类似

的内容
def newPacksGrouped(data: List[Pack]): List[NewPack] =
  data.groupBy(_.name).map{case (name, packs) => NewPack(name, packs)}.toList