在比较和条件检查之后添加列表值-Scala

你的描述仍然很混乱,但我想也许这就是你想要的。

val lst = List(List(1, 0, 0), List(1, 1, 0), List(1, 0, 1))

lst.transpose
   .map(sublst => sublst.sum*0.5 + (if (sublst.forall(_==sublst.head)) 1 else 0))
//res0: List[Double] = List(2.5, 0.5, 0.5)

如果子列表的所有元素都相同,则添加 1 在计算中,否则添加 0 .

lst.transpose.map(v=>if(v.forall(_==v.head)) v.sum*0.5+1 else v.sum*0.5))

或

lst.transpose.map(v=>if(v.exists(_!=v.head)) v.sum*0.5 else v.sum*0.5+1)

或

lst.transpose.map(v=>if(v.find(_!=v.head)!=None) v.sum*0.5 else v.sum*0.5+1)

这些是一般的,可以用来 two different expressions 对两种不同的情况进行评估,即。, all elements equal or not equal 打电话后在名单上 transpose 功能,即。,

lst.transpose.map(v=>if(v.exists(_!=v.head)) 
                     (Expression1 in terms of elements in v) 
                     else (Expression2 in terms of elements in v) )

在Scala REPL中: 因为,
val lst = List(List(1, 0, 0), List(1, 1, 0), List(1, 0, 1))

scala> lst.transpose.map(v=>if(v.forall(_==v.head)) v.sum*0.5+1 else v.sum*0.5)
res23: List[Double] = List(2.5, 0.5, 0.5)

scala> lst.transpose.map(v=>if(v.exists(_!=v.head)) v.sum*0.5 else v.sum*0.5+1)
res24: List[Double] = List(2.5, 0.5, 0.5)

scala> lst.transpose.map(v=>if(v.find(_!=v.head)!=None) v.sum*0.5 else v.sum*0.5+1)
res25: List[Double] = List(2.5, 0.5, 0.5)

可以使用地图和模式匹配

val list = List(List(1, 0, 0), List(1, 1, 0), List(1, 0, 1))

val result = list.transpose.map {
  case sublist if sublist.forall(sublist.head == _) => 1+0.5*sublist.sum
  case sublist => 0.5*sublist.sum
}