Question

I need to find the averages for all columns in the same row of a particular value of the first column. So for a table like the following:

0   11  12  40
1   22  24  92
0   12  13  45
1   24  26  90
2   33  36  138
1   22  24  80
2   36  39  135
0   11  12  46
2   33  36  120

I want the resulting table of averages:

0   11.33333333 12.33333333 43.66666667     
1   22.66666667 24.66666667 87.33333333     
2   34  37  131

So for example for the value 0..
I take the averages of (11 + 12 + 11)/3 on the second column, (12 + 13 + 12)/3 on the second, and (46 + 45 + 40)/3 on the third column

Answer 1

data = "0   11  12  40
        1   22  24  92
        0   12  13  45
        1   24  26  90
        2   33  36  138
        1   22  24  80
        2   36  39  135
        0   11  12  46
        2   33  36  120"

avgs = data.split(/\n/).map{|d| d.split(/\t/)}.group_by{|d| d[0] }.each{|a,b| b.each(&:shift) }.inject({}){|avg, (k, ar)| avg[k] = ar.inject([0,0,0]){|av,(a,b,c)| av[0]+=a.to_f; av[1]+=b.to_f; av[2]+=c.to_f; av}.map{|e| e/ar.size}; avg}
#=> {"0"=>[11.333333333333332, 12.333333333333332, 43.66666666666667], "1"=>[22.666666666666664, 24.666666666666664, 87.33333333333334], "2"=>[34.0, 37.0, 131.0]}

to print it:

avgs.each{|k, arr| puts [k,*arr].join("\t") }
#=> 0   11.333333333333332  12.333333333333332  43.66666666666667
#=> 1   22.666666666666664  24.666666666666664  87.33333333333334
#=> 2   34.0    37.0    131.0

UPD

I’ve cleaned my method a little:

avgs = data.split(/\n/).
            map{|d| d.split(/\t/).map(&:to_f)}.
            group_by(&:first).
            inject({}){|avg, (k, ar)| avg[k] = ar.transpose[1..-1].map{|av| av.inject(:+)/av.size}; avg}