我目前正在学习函数式编程和F#,我想在n-2之前进行循环控制。例如:
Given a list of doubles, find the pairwise average,
e.g. pairwiseAverage [1.0; 2.0; 3.0; 4.0; 5.0] will give [1.5; 2.5; 3.5; 4.5]
在做了一些实验和搜索之后,我有几种方法可以做到:
方法1:
let pairwiseAverage (data: List<double>) =
[for j in 0 .. data.Length-2 do
yield (data.[j]+data.[j+1])/2.0]
方法2:
let pairwiseAverage (data: List<double>) =
let averageWithNone acc next =
match acc with
| (_,None) -> ([],Some(next))
| (result,Some prev) -> ((prev+next)/2.0)::result,Some(next))
let resultTuple = List.fold averageWithNone ([],None) data
match resultTuple with
| (x,_) -> List.rev x
方法3:
let pairwiseAverage (data: List<double>) =
// Get elements from 1 .. n-1
let after = List.tail data
// Get elements from 0 .. n-2
let before =
data |> List.rev
|> List.tail
|> List.rev
List.map2 (fun x y -> (x+y)/2.0) before after
我只想知道是否有其他方法可以解决这个问题。谢谢。
答案 0 :(得分:9)
仅使用内置插件:
{{1}}
Seq.windowed n为您提供每个n个元素的滑动窗口。
答案 1 :(得分:7)
另一种简单的方法是使用Seq.pairwise
类似
list |> Seq.pairwise |> Seq.map (fun (a,b) -> (a+b)/2.0)
答案 2 :(得分:3)
上面提出的方法适用于短窗口,例如问题中的窗口。对于长度大于2的窗户,不能成对使用。 hlo
的答案推广到更宽的窗口,如果窗口长度不是太大,则是一种干净而快速的方法。对于非常宽的窗口,下面的代码运行得更快,因为它只添加一个数字并从前一个窗口获得的值中减去另一个数字。请注意,Seq.map2(和Seq.map)会自动处理不同长度的序列。
let movingAverage (n: int) (xs: float List) =
let init = xs |> (Seq.take n) |> Seq.sum
let additions = Seq.map2 (fun x y -> x - y) (Seq.skip n xs) xs
Seq.fold (fun m x -> ((List.head m) + x)::m) [init] additions
|> List.rev
|> List.map (fun (x: float) -> x/(float n))
xs = [1.0..1000000.0]
movingAverage 1000 xs
// Real: 00:00:00.265, CPU: 00:00:00.265, GC gen0: 10, gen1: 10, gen2: 0
为了进行比较,上面的函数执行上面的计算比窗口等效函数快60倍:
let windowedAverage (n: int) (xs: float List) =
xs
|> Seq.windowed n
|> Seq.map Array.average
|> Seq.toList
windowedAverage 1000 xs
// Real: 00:00:15.634, CPU: 00:00:15.500, GC gen0: 74, gen1: 74, gen2: 71
我尝试使用foldBack消除List.rev,但没有成功。
答案 3 :(得分:2)
无点的方法:
let pairwiseAverage = List.pairwise >> List.map ((<||) (+) >> (*) 0.5)
的 Online Demo 强>
通常不是更好的方式,而是另一种方式,无论......; - ]