高效的Scala惯用方法可以选择85％的排序值？

发布时间：2020-12-16 09:07:45 所属栏目：安全来源：网络整理

导读：鉴于数字列表不增加,我想在列表中选择前85％的值.这是我目前正在做的事情. scala val a = Array(8.60,6.85,4.91,3.45,2.74,2.06,1.53,0.35,0.28,0.12)a: Array[Double] = Array(8.6,0.12)scala val threshold = a.sum * 0.85threshold: Double = 26.2565scal

鉴于数字列表不增加,我想在列表中选择前85％的值.这是我目前正在做的事情.

scala> val a = Array(8.60,6.85,4.91,3.45,2.74,2.06,1.53,0.35,0.28,0.12)
a: Array[Double] = Array(8.6,0.12)

scala> val threshold = a.sum * 0.85
threshold: Double = 26.2565

scala> val successiveSums = a.tail.foldLeft(Array[Double](a.head)){ case (x,y) => x ++ Array(y + x.last) }
successiveSums: Array[Double] = Array(8.6,15.45,20.36,23.81,26.549999999999997,28.609999999999996,30.139999999999997,30.49,30.77,30.89)

scala> successiveSums.takeWhile( x => x <= threshold )
res40: Array[Double] = Array(8.6,23.81)

scala> val size = successiveSums.takeWhile( x => x <= threshold ).size
size: Int = 4

scala> a.take(size)
res41: Array[Double] = Array(8.6,3.45)

我想改进它

>表现
>代码大小

有什么建议？

解决方法

在代码大小上,考虑这个oneliner,

a.take( a.scanLeft(0.0)(_+_).takeWhile( _ <= a.sum * 0.85 ).size - 1 )

这里scanLeft累积了附加内容.

在性能方面,标记中间值可能有助于不重新计算相同的操作,即

val threshold = a.sum * 0.85
val size = a.scanLeft(0.0)(_+_).takeWhile( _ <= threshold ).size - 1
a.take( size )

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!