Merge pull request #18 from K-Ibadullaev/K-Ibadullaev-patch-1

Update SSA.py

Author: K-Ibadullaev

src/py_ssa_lib/SSA.py

@@ -125,6 +125,8 @@ def fit(self, df, L,ts, decomposition,idx_start_ts, **kwargs ):
                                 
             idx_start_ts:int, the first column index of the data frame where the first nummeric value occours,
                               i.e where the time series start(s). Used to cutoff irrelevant columns from the input data set.
+
+            ts: int, the row of the data set containing a time series
             
 
             Returns
@@ -133,6 +135,7 @@ def fit(self, df, L,ts, decomposition,idx_start_ts, **kwargs ):
         """
 
         self.df = df
+        self.ts = ts
         self.idx_start_ts = idx_start_ts
         self.ts_df = self.df.iloc[ts,self.idx_start_ts:].T
         self.decomposition = decomposition
@@ -338,7 +341,7 @@ def reconstruct_ts(self, idx_chosen_components, return_as_df=False):
 
         ts_rec = self.X_to_TS(chosen_components)
         if return_as_df==True:
-            return pd.DataFrame(columns=self.df.columns, data=np.column_stack([self.df.iloc[:,:self.idx_start_ts].values,ts_rec.T ]))
+            return pd.DataFrame(columns=self.df.columns, data=np.column_stack([self.df.iloc[self.ts,:self.idx_start_ts].values,ts_rec.T ]))
         return ts_rec.T