I have been just summing over and then use reduce at the end of the loop then average at each z index.