From f05afef666fdf771b4e7a1d99c7111920c75b491 Mon Sep 17 00:00:00 2001
From: ViK <bamnup12@gmail.com>
Date: Mon, 3 Nov 2014 11:42:28 +0200
Subject: [PATCH] added weighted Pearson correlation

---
 src/Numerics/Statistics/Correlation.cs | 67 ++++++++++++++++++++++++++
 1 file changed, 67 insertions(+)

diff --git a/src/Numerics/Statistics/Correlation.cs b/src/Numerics/Statistics/Correlation.cs
index f880e504..5a27f4e6 100644
--- a/src/Numerics/Statistics/Correlation.cs
+++ b/src/Numerics/Statistics/Correlation.cs
@@ -96,6 +96,73 @@ namespace MathNet.Numerics.Statistics
             return r / Math.Sqrt(varA * varB);
         }
 
+        /// <summary>
+        /// Computes the Weighted Pearson Product-Moment Correlation coefficient.
+        /// </summary>
+        /// <param name="dataA">Sample data A.</param>
+        /// <param name="dataB">Sample data B.</param>
+        /// <param name="weights">Corresponding weights of data.</param>
+        /// <returns>The Weighted Pearson product-moment correlation coefficient.</returns>
+        public static double WeightedPearson(IEnumerable<double> dataA, IEnumerable<double> dataB, IEnumerable<double> weights)
+        {
+            int n = 0;
+
+            double meanA = 0;
+            double meanB = 0;
+            double varA = 0;
+            double varB = 0;
+            double sumWeight = 0;
+
+            double covariance = 0;
+
+            using (IEnumerator<double> ieA = dataA.GetEnumerator())
+            using (IEnumerator<double> ieB = dataB.GetEnumerator())
+            using (IEnumerator<double> ieW = weights.GetEnumerator())
+            {
+                while (ieA.MoveNext())
+                {
+                    if (!ieB.MoveNext())
+                    {
+                        throw new ArgumentOutOfRangeException("dataB", Resources.ArgumentArraysSameLength);
+                    }
+                    if (!ieW.MoveNext())
+                    {
+                        throw new ArgumentOutOfRangeException("weights", Resources.ArgumentArraysSameLength);
+                    }
+                    ++n;
+
+                    double xi = ieA.Current;
+                    double yi = ieB.Current;
+                    double wi = ieW.Current;
+
+                    double temp = sumWeight + wi;
+
+                    double deltaX = xi - meanA;
+                    double rX = deltaX * wi / temp;
+                    meanA += rX;
+                    varA += sumWeight * deltaX * rX;
+
+                    double deltaY = yi - meanB;
+                    double rY = deltaY * wi / temp;
+                    meanB += rY;
+                    varB += sumWeight * deltaY * rY;
+
+                    sumWeight = temp;
+
+                    covariance += deltaX * deltaY * (n - 1) * wi / n;
+                }
+                if (ieB.MoveNext())
+                {
+                    throw new ArgumentOutOfRangeException("dataB", Resources.ArgumentArraysSameLength);
+                }
+                if (ieW.MoveNext())
+                {
+                    throw new ArgumentOutOfRangeException("weights", Resources.ArgumentArraysSameLength);
+                }
+            }
+            return covariance / Math.Sqrt(varA * varB);
+        }
+
         /// <summary>
         /// Computes the Pearson Product-Moment Correlation matrix.
         /// </summary>