Added string serialization/deserialization
[sandbox] / cryptopals / 01.03 / stat.erl
1 -module(stat).
2 -export([sum/1,mean/1,r_value/1,frequency/1,frequencies_to_scatterplot/2]).
3
4 sum([]) -> 0;
5 sum([Head|Tail]) -> Head + sum(Tail).
6
7 mean(List) -> sum(List) / length(List).
8
9 covariance(Points, MeanX, MeanY) ->
10     sum(lists:map(fun({X,Y}) -> (X - MeanX) * (Y - MeanY) end, Points)).
11
12 standard_deviation(Samples, Mean) ->
13     math:sqrt(sum(lists:map(
14         fun(Item) -> math:pow(Item - Mean, 2) end,
15         Samples))).
16
17 r_value(Points) ->
18     Xs = lists:map(fun({X,_}) -> X end, Points),
19     Ys = lists:map(fun({_,Y}) -> Y end, Points),
20     MeanX = mean(Xs),
21     MeanY = mean(Ys),
22     Covariance = covariance(Points, MeanX, MeanY),
23     StandardDeviationX = standard_deviation(Xs, MeanX),
24     StandardDeviationY = standard_deviation(Ys, MeanY),
25     Covariance / StandardDeviationX / StandardDeviationY.
26
27 frequency([],Result) -> Result;
28 frequency([Head|Tail],Result) ->
29     frequency(Tail,dict:update_counter(Head,1,Result)).
30
31 frequency(Sample) -> frequency(Sample, dict:new()).
32
33 fetch_values(Dict) -> lists:map(
34     fun({_,Value}) -> Value end,
35     dict:to_list(Dict)).
36
37 frequencies_to_scatterplot(Fx,Fy) ->
38     FxPoints = dict:map(fun(_,Value) -> {Value,0} end, Fx),
39     FyPoints = dict:map(fun(_,Value) -> {0,Value} end, Fy),
40     KeysToPoints = dict:merge(
41         fun(_,{X,_},{_,Y}) -> {X,Y} end,
42         FxPoints,
43         FyPoints),
44     fetch_values(KeysToPoints).