2009-11-27 7 views
13

Supponiamo che io sono un elenco di tupples come questi:gruppo vicino con tuple in F #

[("A",12); ("A",10); ("B",1); ("C",2); ("C",1)] 

E mi piacerebbe fare una sorta di groupby come faccio a gestire questo?

In pseudocodice-SQL che dovrebbe essere simile a questa:

SELECT fst(tpl), sum(lst(tpl)) FROM [TupplesInList] GROUP BY fst(tpl) 

cedendo

[("A",22); ("B",1); ("C",3)] 

potrei fare un dizionario e aggiungere le interi se la chiave esiste, ma faccio fatica a credere che sarebbe la migliore soluzione in una lingua così espressiva come F #.

risposta

24

Una soluzione:

let tuples = [("A",12); ("A",10); ("B",1); ("C",2); ("C",1)] 
tuples 
|> Seq.groupBy fst 
|> Seq.map (fun (key, values) -> (key, values |> Seq.sumBy snd)) 

Edit: ... o senza tubazioni:

let tuples = [("A",12); ("A",10); ("B",1); ("C",2); ("C",1)] 
Seq.map (fun (key, group) -> key, Seq.sumBy snd group) 
     (Seq.groupBy fst tuples) 
14

Per espandere sulla risposta di Johan, tendo a fare questa cosa tipo molto e così hanno fatto la seguente funzione generalizzata.

let group_fold key value fold acc seq = 
    seq |> Seq.groupBy key 
     |> Seq.map (fun (key, seq) -> (key, seq |> Seq.map value |> Seq.fold fold acc)) 

che lavora per il vostro caso tuple, come si vede qui sotto

let tuples = [("A",12); ("A",10); ("B",1); ("C",2); ("C",1)] 

let regular = group_fold fst snd (+) 0 tuples 
let piped = tuples |> group_fold fst snd (+) 0 

ma funziona anche con altri seqences come una lista di stringhe

let strings = ["A12"; "A10"; "B1"; "C2"; "C1"] 

let regular = group_fold (fun (x : string) -> x.[0]) (fun (x : string) -> int x.[1..]) (+) 0 strings 
let piped = strings |> group_fold (fun x -> x.[0]) (fun x -> int x.[1..]) (+) 0