@@ -105,21 +105,15 @@ program demo_moment
105
105
use stdlib_experimental_stats, only: mean, moment
106
106
implicit none
107
107
real :: x(1:6) = [ 1., 2., 3., 4., 5., 6. ]
108
- print *, moment(x, 2, center = mean(x))
109
- !returns 2.9167
110
-
111
- print *, moment( reshape(x, [ 2, 3 ] ), 2,&
112
- center = mean(reshape(x, [ 2, 3 ])))
113
- !returns 2.9167
114
-
115
- print *, moment( reshape(x, [ 2, 3 ] ), 2, 1,&
116
- center = mean(reshape(x, [ 2, 3 ] ), 1))
117
- !returns [0.25, 0.25, 0.25]
118
-
119
- print *, moment( reshape(x, [ 2, 3 ] ), 2, 1,&
120
- center = mean(reshape(x, [2, 3]), 1, reshape(x, [ 2, 3 ] ) > 3.),&
121
- mask = (reshape(x, [ 2, 3 ] ) > 3.))
122
- !returns [NaN, 0., 0.25]
108
+ real :: y(1:2, 1:3) = reshape([ 1., 2., 3., 4., 5., 6. ], [ 2, 3])
109
+ print *, moment(x, 2, center = mean(x)) !returns 2.9167
110
+ print *, moment( y, 2,&
111
+ center = mean(y)) !returns 2.9167
112
+ print *, moment( y, 2, 1,&
113
+ center = mean(y, 1)) !returns [0.25, 0.25, 0.25]
114
+ print *, moment( y, 2, 1,&
115
+ center = mean(y, 1, y > 3.),&
116
+ mask = (y > 3.)) !returns [NaN, 0., 0.25]
123
117
end program demo_moment
124
118
```
125
119
0 commit comments