Advertisement
Guest User

Untitled

a guest
Sep 21st, 2017
65
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.65 KB | None | 0 0
  1. >>> x = np.arange(0,10)
  2. >>> y1 = np.exp(-x/3.0)
  3. >>> y2 = np.sqrt(x)
  4. >>> df = pd.DataFrame(np.concatenate((y1[:,np.newaxis], y2[:,np.newaxis]),axis=1), index=x)
  5.  
  6. >>> f = [scipy.interpolate.interp1d(df.index, df.iloc[:, i], kind='linear') for i in range(2)]
  7.  
  8. class Test:
  9.  
  10. def __init__(self, df):
  11. self.df = df
  12. self._intep = [scipy.interpolate.interp1d(df.index,df.iloc[:, i], kind='linear') for i in range(len(df.columns))]
  13.  
  14. def __getitem__(self, pos):
  15. x, y = pos
  16. col = self.df.columns[y]
  17. l = [self._intep[i](x) for i in range(len(col))]
  18. return np.column_stack(l)
  19.  
  20. t = Test(df)
  21. t[[1, 2, 4], :]
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement