Ottenere ogni ennesimo elemento di una sequenza

Question

Sto cercando un modo per creare una sequenza composta da ogni ennesimo elemento di un'altra sequenza, ma non sembra trovare un modo per farlo in modo elegante. Ovviamente posso modificare qualcosa, ma mi chiedo se esiste una funzione di libreria che non vedo.

Le funzioni di sequenza i cui nomi finiscono in -i sembrano essere abbastanza buona con lo scopo di capire quando un elemento è l'ennesimo o (multiplo di n) esimo uno, ma posso vedere solo iteri e mapi, nessuno di cui si presta davvero al compito.

Esempio:

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

Poi partial dovrebbe essere [3;6]. C'è qualcosa di simile là fuori?

Edit:

Se io non sono abbastanza ambiziosi e consentire la n essere costante/conosciuto, poi ho appena trovato che il seguente dovrebbe funzionare:

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

Would ci sarà un modo per scrivere questo più breve?

Answer 1

È possibile ottenere il comportamento componendo mapi con le altre funzioni:

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 

la soluzione con opzioni e choose come suggerito da Annone userebbe solo due funzioni, ma il corpo del primo sarebbe un po 'più complicato (ma il principio è essenzialmente lo stesso).

una versione più efficiente utilizzando direttamente l'oggetto IEnumerator non è troppo difficile da scrivere:

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 } 

EDIT: Il frammento è anche disponibile qui: http://fssnip.net/1R

Answer 2

Seq.choose funziona bene in queste situazioni, perché consente di eseguire il lavoro filter all'interno del lambda mapi.

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

Simile a here.