Update xgi/linalg/laplacian_matrix.py

Implement suggestion - Edge weight matrix Co-authored-by: Maxime Lucas <[email protected]>
xgi-org · Jan 28, 2025 · 5de2a16 · 5de2a16
1 parent 4881074
commit 5de2a16
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/xgi/linalg/laplacian_matrix.py b/xgi/linalg/laplacian_matrix.py
@@ -270,12 +270,8 @@ def normalized_hypergraph_laplacian(H, weights=None, sparse=True, index=False):
     else:
         Dv_invsqrt = np.diag(np.power(Dv, -0.5))
 
-        De_inv = np.diag(list(
-            map(
-                lambda x: 1/x,
-                np.sum(H_, axis=0)
-            )
-        ))
+        De = np.sum(H_, axis=0)
+        De_inv = np.diag(1 / De)
 
         if weights is not None:
             W = np.diag(weights)