获取序列的每 n 个元素

Question

我正在寻找一种方法来创建一个由另一个序列的每个第 n 个元素组成的序列，但似乎没有找到一种优雅的方法来做到这一点。我当然可以破解一些东西，但我想知道是否有我没有看到的库函数。

名称以 -i 结尾的序列函数似乎非常适合确定元素何时是第 n 个或（n 的倍数）第一个，但我只能看到 iteri 和 @987654322 @，没有一个真正适合这项任务。

例子：

let someseq = [1;2;3;4;5;6]
let partial = Seq.magicfunction 3 someseq

那么partial 应该是[3;6]。那里有类似的东西吗？

编辑：

如果我不那么雄心勃勃并允许n 保持不变/已知，那么我刚刚发现以下应该可行：

let rec thirds lst =
    match lst with
    | _::_::x::t -> x::thirds t // corrected after Tomas' comment
    | _ -> []

有没有办法把这个写得更短？

Answer 1

Seq.choose 在这些情况下工作得很好，因为它允许您在 mapi lambda 中执行 filter 工作。

let everyNth n elements =
    elements
    |> Seq.mapi (fun i e -> if i % n = n - 1 then Some(e) else None)
    |> Seq.choose id

类似于here。

Answer 2

您可以通过将mapi 与其他函数组合来获得该行为：

let everyNth n seq = 
  seq |> Seq.mapi (fun i el -> el, i)              // Add index to element
      |> Seq.filter (fun (el, i) -> i % n = n - 1) // Take every nth element
      |> Seq.map fst                               // Drop index from the result

Annon 建议的使用 options 和 choose 的解决方案将只使用两个函数，但第一个函数的主体会稍微复杂一些（但原理基本相同）。

直接使用IEnumerator 对象的更高效的版本并不难写：

let everyNth n (input:seq<_>) = 
  seq { use en = input.GetEnumerator()
        // Call MoveNext at most 'n' times (or return false earlier)
        let rec nextN n = 
          if n = 0 then true
          else en.MoveNext() && (nextN (n - 1)) 
        // While we can move n elements forward...
        while nextN n do
          // Retrun each nth element
          yield en.Current }

编辑：sn-p 也可在此处获得：http://fssnip.net/1R