vecvec at mat arr means

Author: ChrisRackauckas

src/RecursiveArrayTools.jl

@@ -2,6 +2,8 @@ __precompile__()
 
 module RecursiveArrayTools
 
+  import Base: mean
+
   function recursivecopy!{T<:Number,N}(b::Array{T,N},a::Array{T,N})
     @inbounds copy!(b,a)
   end
@@ -54,7 +56,29 @@ module RecursiveArrayTools
 
   recursive_one(a) = recursive_one(a[1])
   recursive_one{T<:Number}(a::T) = one(a)
+  
+  function mean{T<:AbstractArray}(vecvec::Vector{T})
+    out = zeros(vecvec[1])
+    for i in eachindex(vecvec)
+      out+= vecvec[i]
+    end
+    out/length(vecvec)
+  end
+
+  function mean{T<:AbstractArray}(matarr::Matrix{T},region=0)
+    if region == 0
+      return mean(vec(matarr))
+    elseif region == 1
+      out = [zeros(matarr[1,i]) for i in 1:size(matarr,2)]
+      for j in 1:size(matarr,2), i in 1:size(matarr,1)
+        out[j] += matarr[i,j]
+      end
+      return out/size(matarr,1)
+    elseif region == 2
+      return mean(matarr',1)
+    end
+  end
 
-  export recursivecopy!, vecvecapply, copyat_or_push!, vecvec_to_mat, recursive_one
+  export recursivecopy!, vecvecapply, copyat_or_push!, vecvec_to_mat, recursive_one,mean
 
 end # module

test/runtests.jl

@@ -19,3 +19,20 @@ t = collect(linspace(0,10,200))
 randomized = [(sol(t[i]) + .01randn(2)) for i in 1:length(t)]
 data = vecvec_to_mat(randomized)
 @test typeof(data) <: Matrix{Float64}
+
+## Test means
+A = [[1 2; 3 4],[1 3;4 6],[5 6;7 8]]
+@test mean(A) ≈ [2.33333333 3.666666666
+           4.6666666666 6.0]
+B = Matrix{Matrix{Int64}}(2,3)
+B[1,:] = [[1 2; 3 4],[1 3;4 6],[5 6;7 8]]
+B[2,:] = [[1 2; 3 4],[1 5;4 3],[5 8;2 1]]
+
+ans = [[1 2; 3 4],[1 4; 4 4.5],[5 7; 4.5 4.5]]
+@test mean(B,1)[1] ≈ ans[1]
+@test mean(B,1)[2] ≈ ans[2]
+@test mean(B,1)[3] ≈ ans[3]
+
+ans = [[2.333333333333 4.666666666666; 3.6666666666666 6.0], [2.3333333 3.0; 5.0 2.6666666]]
+@test mean(B,2)[1] ≈ ans[1]
+@test mean(B,2)[2] ≈ ans[2]