column math

[scidam]

You don't really need to initialize direction column.

data.loc[data.Goal.shift(5) - data.Last >= 5, 'direction'] = 1
data.loc[data.Goal.shift(5) - data.Last <= -5, 'direction'] = -1
data.direction.fillna(0, inplace=True) # optionally, instead of initialization