gsl_statistics__variance_source.c 2.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091
  1. /* statistics/variance_source.c
  2. *
  3. * Copyright (C) 1996, 1997, 1998, 1999, 2000, 2007 Jim Davies, Brian Gough
  4. *
  5. * This program is free software; you can redistribute it and/or modify
  6. * it under the terms of the GNU General Public License as published by
  7. * the Free Software Foundation; either version 3 of the License, or (at
  8. * your option) any later version.
  9. *
  10. * This program is distributed in the hope that it will be useful, but
  11. * WITHOUT ANY WARRANTY; without even the implied warranty of
  12. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  13. * General Public License for more details.
  14. *
  15. * You should have received a copy of the GNU General Public License
  16. * along with this program; if not, write to the Free Software
  17. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  18. */
  19. static double
  20. FUNCTION(compute,variance) (const BASE data[], const size_t stride, const size_t n, const double mean);
  21. static double
  22. FUNCTION(compute,variance) (const BASE data[], const size_t stride, const size_t n, const double mean)
  23. {
  24. /* takes a dataset and finds the variance */
  25. long double variance = 0 ;
  26. size_t i;
  27. /* find the sum of the squares */
  28. for (i = 0; i < n; i++)
  29. {
  30. const long double delta = (data[i * stride] - mean);
  31. variance += (delta * delta - variance) / (i + 1);
  32. }
  33. return variance ;
  34. }
  35. double
  36. FUNCTION(gsl_stats,variance_with_fixed_mean) (const BASE data[], const size_t stride, const size_t n, const double mean)
  37. {
  38. const double variance = FUNCTION(compute,variance) (data, stride, n, mean);
  39. return variance;
  40. }
  41. double
  42. FUNCTION(gsl_stats,sd_with_fixed_mean) (const BASE data[], const size_t stride, const size_t n, const double mean)
  43. {
  44. const double variance = FUNCTION(compute,variance) (data, stride, n, mean);
  45. const double sd = sqrt (variance);
  46. return sd;
  47. }
  48. double
  49. FUNCTION(gsl_stats,variance_m) (const BASE data[], const size_t stride, const size_t n, const double mean)
  50. {
  51. const double variance = FUNCTION(compute,variance) (data, stride, n, mean);
  52. return variance * ((double)n / (double)(n - 1));
  53. }
  54. double
  55. FUNCTION(gsl_stats,sd_m) (const BASE data[], const size_t stride, const size_t n, const double mean)
  56. {
  57. const double variance = FUNCTION(compute,variance) (data, stride, n, mean);
  58. const double sd = sqrt (variance * ((double)n / (double)(n - 1)));
  59. return sd;
  60. }
  61. double
  62. FUNCTION(gsl_stats,variance) (const BASE data[], const size_t stride, const size_t n)
  63. {
  64. const double mean = FUNCTION(gsl_stats,mean) (data, stride, n);
  65. return FUNCTION(gsl_stats,variance_m)(data, stride, n, mean);
  66. }
  67. double
  68. FUNCTION(gsl_stats,sd) (const BASE data[], const size_t stride, const size_t n)
  69. {
  70. const double mean = FUNCTION(gsl_stats,mean) (data, stride, n);
  71. return FUNCTION(gsl_stats,sd_m) (data, stride, n, mean);
  72. }