Fixed non-integer array indexing

mims-harvard · Sep 22, 2017 · fa0f3d2 · fa0f3d2
1 parent be02737
commit fa0f3d2
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/nimfa/methods/factorization/psmf.py b/nimfa/methods/factorization/psmf.py
@@ -336,8 +336,8 @@ def _update_lamb(self):
                 locs = sub2ind(
                     (self.V.shape[0], self.rank), locs, self.s[locs, n])
                 for l in locs:
-                    self.W[l % self.V.shape[0], l / self.V.shape[0]] = self.lamb[
-                        l % self.V.shape[0], l / self.V.shape[0]]
+                    self.W[l % self.V.shape[0], l // self.V.shape[0]] = self.lamb[
+                        l % self.V.shape[0], l // self.V.shape[0]]
         self.W = self.W.tocsr()
         self._cross_terms()
 

diff --git a/nimfa/methods/factorization/snmf.py b/nimfa/methods/factorization/snmf.py
@@ -403,7 +403,7 @@ def _spfcnnls(self, C, A):
                                 tmp_f[i, h_set[j]] = True
                     idx_f = find(tmp_f[:, h_set])
                     i_f = [l % p_set.shape[0] for l in idx_f]
-                    j_f = [l / p_set.shape[0] for l in idx_f]
+                    j_f = [l // p_set.shape[0] for l in idx_f]
                     if len(i_f) == 0:
                         break
                     if n_h_set == 1:

diff --git a/nimfa/utils/linalg.py b/nimfa/utils/linalg.py
@@ -279,7 +279,7 @@ def _naxis(row, col):
     else:
         idxX = np.asmatrix(X).argmax(axis)
         if axis is None:
-            eX = X[idxX / X.shape[1], idxX % X.shape[1]]
+            eX = X[idxX // X.shape[1], idxX % X.shape[1]]
         elif axis == 0:
             eX = [X[idxX[0, idx], col]
                   for idx, col in zip(range(X.shape[1]), range(X.shape[1]))]
@@ -333,7 +333,7 @@ def _naxis(row, col):
     else:
         idxX = np.asmatrix(X).argmin(axis)
         if axis is None:
-            eX = X[idxX / X.shape[1], idxX % X.shape[1]]
+            eX = X[idxX // X.shape[1], idxX % X.shape[1]]
         elif axis == 0:
             eX = [X[idxX[0, idx], col]
                   for idx, col in zip(range(X.shape[1]), range(X.shape[1]))]