From 77bcf836272faf1a40fdea97131b85390e935be3 Mon Sep 17 00:00:00 2001 From: Rémi Flamary Date: Tue, 20 Jun 2017 14:51:12 +0200 Subject: add clean zeros function for sparse distributions --- ot/utils.py | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'ot/utils.py') diff --git a/ot/utils.py b/ot/utils.py index fc6b0d2..7ad7637 100644 --- a/ot/utils.py +++ b/ot/utils.py @@ -50,6 +50,13 @@ def unif(n): """ return np.ones((n,))/n +def clean_zeros(a,b,M): + """ Remove all components with zeros weights in a and b + """ + M2=M[a>0,:][:,b>0].copy() # copy force c style matrix (froemd) + a2=a[a>0] + b2=b[b>0] + return a2,b2,M2 def dist(x1,x2=None,metric='sqeuclidean'): """Compute distance between samples in x1 and x2 using function scipy.spatial.distance.cdist -- cgit v1.2.3