Mercurial Hosting > traffic-intelligence
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'