Update CoreTraidMath.py

This commit is contained in:
Redsandy 2022-03-04 16:21:28 +03:00 committed by GitHub
parent fabc61302a
commit 13232c07ca

View File

@ -29,13 +29,6 @@ class CoreMath:
def __init__(self, base_df, params={ def __init__(self, base_df, params={
'dataType':'ohcl', 'dataType':'ohcl',
'colName':{
'open':'open',
'close':'close',
'high':'high',
'low':'low',
'date':'date'
},
'action': None, 'action': None,
'actionOptions':{} 'actionOptions':{}
} }
@ -44,7 +37,7 @@ class CoreMath:
self.base_df=base_df.reset_index(drop=True) self.base_df=base_df.reset_index(drop=True)
self.params=params self.params=params
if self.params['dataType']=='ohcl': if self.params['dataType']=='ohcl':
self.col=self.base_df[self.params['colName'][self.params['actionOptions']['valueType']]] self.col=self.base_df[[self.params['actionOptions']['valueType']]]
elif self.params['dataType']=='series': elif self.params['dataType']=='series':
self.col=self.base_df self.col=self.base_df
self.ans=self.getAns() self.ans=self.getAns()
@ -121,4 +114,5 @@ class CoreMath:
return(ans) return(ans)