Mercurial > repos > sauria > create_correlation_matrix

changeset 3:89009e9b7eb0 draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Uploaded
author sauria
date Thu, 27 Apr 2017 17:28:27 -0400
parents f0c8cdd78e28
children 4a76c97c3dd0
files correlation_matrix.py
diffstat 1 files changed, 7 insertions(+), 3 deletions(-) [+]
line wrap: on
line diff
--- a/correlation_matrix.py	Thu Apr 27 12:37:59 2017 -0400
+++ b/correlation_matrix.py	Thu Apr 27 17:28:27 2017 -0400
@@ -8,16 +8,19 @@
 def main():
     parser = generate_parser()
     args = parser.parse_args()
-    data, names = load_data(args)
+    data, rnames = load_data(args)
     corr = find_correlations(data, args)
     save_data(corr, names, args)
 
 def load_data(args):
     infile = open(args.input)
     names = []
+    cnames = None
     data = []
     if args.column:
-        temp = infile.readline()
+        cnames = infile.readline().split()
+        if args.rows:
+            cnames = cnames[1:]
     temp = infile.readline()
     if args.int:
         dtype = int
@@ -37,7 +40,8 @@
     data = numpy.array(data)
     if args.features:
         data = data.T
-    return data, names
+        names = cnames
+    return data, names, cnames
 
 def find_correlations(data, args):
     corr = numpy.ones((data.shape[0], data.shape[0]), dtype=numpy.float32)