Work with the following data:
import datetime,numpy as np,pandas as pd
nan = np.nan
a = pd.DataFrame( {'price': {datetime.time(9, 0): 1, datetime.time(10, 0): 0, datetime.time(11, 0): 3, datetime.time(12, 0): 4, datetime.time(13, 0): 7, datetime.time(14, 0): 6, datetime.time(15, 0): 5, datetime.time(16, 0): 4, datetime.time(17, 0): 0, datetime.time(18, 0): 2, datetime.time(19, 0): 4, datetime.time(20, 0): 7}, 'reversal': {datetime.time(9, 0): 1, datetime.time(10, 0): nan, datetime.time(11, 0): nan, datetime.time(12, 0): nan, datetime.time(13, 0): nan,
datetime.time(14, 0): 6.0, datetime.time(15, 0): nan, datetime.time(16, 0): nan, datetime.time(17, 0): nan, datetime.time(18, 0): nan, datetime.time(19, 0): nan, datetime.time(20, 0): nan}})
a['target_hit']=nan;
a['target_miss']=nan;
a['reversal1']=a['reversal']+1;
a['reversal2']=a['reversal']-a['reversal'];
a.sort_index(1,inplace=True);
I create a subset as follows:
hit = a.ix[:,:-2].dropna()
hit
Which gives a result showing two lines that match:
price reversal reversal1 reversal2
09:00:00 1 1.0 2.0 0.0
14:00:00 6 6.0 7.0 0.0
When I try to use these lines to match the following, I get this error ValueError: operands could not be broadcast together with shapes (2,) (12,)
takeBoth = False
targetIsHit,targetIsMiss = False,False
if takeBoth:
targetHit = a[(hit['reversal1'].values==a['price'].values) & (hit['reversal1'].index.values<a['price'].index.values)];
targetMiss = a[(hit['reversal2'].values==a['price'].values) & (hit['reversal2'].index.values<a['price'].index.values)];
targetIsHit,targetIsMiss = not targetHit.empty, not targetMiss.empty
else:
targetHit = a[(hit['reversal1'].values==a['price'].values) & (hit['reversal1'].index.values<a['price'].index.values)];
targetIsHit = not targetHit.empty
if not targetIsHit:
targetMiss = a[(hit['reversal2'].values==a['price'].values) & (hit['reversal2'].index.values<a['price'].index.values)];
targetIsMiss = not targetMiss.empty
if targetIsHit:a.loc[hit.index.values,"target_hit"] = targetHit.index.values;
if targetIsMiss:a.loc[hit.index.values,"target_miss"] = targetMiss.index.values;
I do not get this error if it hit = a.ix[:,:-2].dropna()creates only one line. After reading about this, I see that this is possible due to the broadcasting rule.
Should I repeat line by line in hitorder to avoid this? Any other suggestions on how to fix this, please?