import numpy as np import pandas as pd data = pd.DataFrame(np.random.rand(6, 1), columns=['close'], index=['a', 'b', 'c', 'd', 'e', 'f']) data['shift_plus_1'] = data.close.shift(1) data.loc[data.close.shift(+1) < data.close, 'TrendUp'] = 1 print(data)
Результат выполнения скрипта
close shift_plus_1 TrendUp a 0.743436 NaN NaN b 0.254870 0.743436 NaN c 0.689602 0.254870 1.0 d 0.233681 0.689602 NaN e 0.100933 0.233681 NaN f 0.391622 0.100933 1.0
Leave a Reply