Merge pull request #2752 from aszilagyi/patch-1

tacaswell · tacaswell · commit 051145077f74 · 2014-01-29T16:11:06.000-08:00
Make standardization of input optional in mlab.PCA
diff --git a/lib/matplotlib/mlab.py b/lib/matplotlib/mlab.py
@@ -1653,14 +1653,16 @@ def prepca(P, frac=0):
 
 
 class PCA:
-    def __init__(self, a):
+    def __init__(self, a, standardize=True):
         """
         compute the SVD of a and store data for PCA.  Use project to
         project the data onto a reduced set of dimensions
 
         Inputs:
 
           *a*: a numobservations x numdims array
+          *standardize*: True if input data are to be standardized. If False, only centering will be
+          carried out.
 
         Attrs:
 
@@ -1694,6 +1696,7 @@ def __init__(self, a):
         self.numrows, self.numcols = n, m
         self.mu = a.mean(axis=0)
         self.sigma = a.std(axis=0)
+        self.standardize = standardize
 
         a = self.center(a)
 
@@ -1745,8 +1748,11 @@ def project(self, x, minfrac=0.):
 
 
     def center(self, x):
-        'center the data using the mean and sigma from training set a'
-        return (x - self.mu)/self.sigma
+        'center and optionally standardize the data using the mean and sigma from training set a'
+        if self.standardize:
+            return (x - self.mu)/self.sigma
+        else:
+            return (x - self.mu)
 
 
 
