Method data_process

finagent/tools/strategy_agents.py:11–21

Method to process the input dataframe. Renames columns to standard names.

(self, data)

    """

    def data_process(self, data):
        """
        Method to process the input dataframe.
        Renames columns to standard names.
        """
        if isinstance(data, np.ndarray):
            df = pd.DataFrame(data, columns=['open', 'high', 'low', 'close', 'adj_close'])
        else:
            df = data
        df = df.rename(columns={'open': 'Open', 'high': 'High', 'low': 'Low', 'close': 'Close', 'volume': 'Volume'})
        return df

    def wrapper(self,strategy, data,params=None):
        """