(self, num_values, min_periods, center, closed, step)
| 47 | |
| 48 | class CustomIndexer(BaseIndexer): |
| 49 | def get_window_bounds(self, num_values, min_periods, center, closed, step): |
| 50 | start = np.empty(num_values, dtype=np.int64) |
| 51 | end = np.empty(num_values, dtype=np.int64) |
| 52 | for i in range(num_values): |
| 53 | if self.use_expanding[i]: |
| 54 | start[i] = 0 |
| 55 | end[i] = i + 1 |
| 56 | else: |
| 57 | start[i] = i |
| 58 | end[i] = i + self.window_size |
| 59 | return start, end |
| 60 | |
| 61 | indexer = CustomIndexer(window_size=1, use_expanding=use_expanding) |
| 62 | result = df.rolling(indexer).sum() |