changeset 841:90b7d6e19c55

minor
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Mon, 11 Jul 2016 17:46:00 -0400
parents 15a82ebc62c4
children 75530d8c0090
files python/utils.py
diffstat 1 files changed, 6 insertions(+), 2 deletions(-) [+]
line wrap: on
line diff
--- a/python/utils.py	Fri Jul 08 11:41:29 2016 -0400
+++ b/python/utils.py	Mon Jul 11 17:46:00 2016 -0400
@@ -484,9 +484,13 @@
         out.close()
     return result
 
-def saveDokMatrix(filename, m):
+def saveDokMatrix(filename, m, lowerTriangle = False):
     'Saves a dok_matrix using savez'
-    savez(filename, shape = m.shape, keys = m.keys(), values = m.values())
+    if lowerTriangle:
+        keys = [k for k in m.keys() if k[0] > k[1]]
+        savez(filename, shape = m.shape, keys = keys, values = [m[k[0],k[1]] for k in keys])
+    else:
+        savez(filename, shape = m.shape, keys = m.keys(), values = m.values())
 
 def loadDokMatrix(filename):
     'Loads a dok_matrix saved using the above saveDokMatrix'