diff options
Diffstat (limited to 'series_expansion.py')
-rw-r--r-- | series_expansion.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/series_expansion.py b/series_expansion.py index 15ee79d..ca06b40 100644 --- a/series_expansion.py +++ b/series_expansion.py @@ -59,7 +59,7 @@ class SeriesExpansion(object): basis_vals = [] for i, (b, c, d) in enumerate(zip(self._basis, coords, diff_order)): val = [] - for idx in xrange(self._coeffs.shape[i]): + for idx in range(self._coeffs.shape[i]): val.append(b.eval(idx, c, d)) basis_vals.append(val) @@ -68,8 +68,8 @@ class SeriesExpansion(object): ret += val * c return ret elif self._dim == 2: - for i in xrange(self._coeffs.shape[0]): - for j in xrange(self._coeffs.shape[1]): + for i in range(self._coeffs.shape[0]): + for j in range(self._coeffs.shape[1]): ret += self._coeffs[i, j] * np.outer(basis_vals[0][i], basis_vals[1][j]) return ret else: |