在元组字段上应用部分函数,​​维护元组结构

时间:2016-05-16 06:21:09

标签: scala functional-programming partialfunction

我有PartialFunction[String,String]Map[String,String]。 我想在地图值上应用部分函数并收集它所适用的条目。 即给定:

val m = Map( "a"->"1", "b"->"2" )
val pf : PartialFunction[String,String] = {
  case "1" => "11"
}

我想以某种方式将_._2pf合并,并且能够做到这一点:

val composedPf : PartialFunction[(String,String),(String,String)] = /*someMagicalOperator(_._2,pf)*/
val collected : Map[String,String] = m.collect( composedPf ) 
// collected should be Map( "a"->"11" )
到目前为止,我得到的最好的是:

val composedPf = new PartialFunction[(String,String),(String,String)]{
        override def isDefinedAt(x: (String, String)): Boolean = pf.isDefinedAt(x._2)
        override def apply(v1: (String, String)): (String,String) = v1._1 -> pf(v1._2)
      }

有更好的方法吗?

2 个答案:

答案 0 :(得分:3)

这是神奇的操作员:

val composedPf: PartialFunction[(String, String), (String, String)] = 
  {case (k, v) if pf.isDefinedAt(v) => (k, pf(v))}

另一个没有创建组合函数的选项是:

m.filter(e => pf.isDefinedAt(e._2)).mapValues(pf)

答案 1 :(得分:0)

Scalaz中有一个功能就是这样:second

scala> m collect pf.second
res0: scala.collection.immutable.Map[String,String] = Map(a -> 11)

这是有效的,因为PartialFunction是箭头(一般化函数)类型类的实例,而second是为箭头定义的常见操作之一。