Как добавить условия с скользящими средними значениями дня в пандах? - PullRequest
Новая
       20

Как добавить условия с скользящими средними значениями дня в пандах?

0 голосов
/ 28 января 2019

Домашнее задание. Нужна помощь в решении: стратегия: покупайте, когда цена поднимается выше 50-дневной скользящей средней, а затем продавайте после 3 торговых сессий.Какую прибыль (в%) мы бы получили в среднем?В торговый день x мы говорим, что цена «идет выше» 50-дневной скользящей средней, если (1) цена была ниже скользящей средней в торговый день x-1 и (2) цена выше скользящей средней в торговледень x.

Как включить условие [1] и [2] в мой текущий код с помощью скользящей средней за 50 дней: 

rol=stock.rolling(50).mean()

profitMade=((stock.shift(-3)-stock)/stock)

stock>rol

profitMade[stock>rol].mean()

Образец набора данных: акции и соответствующая дата: 

Date
2002-05-23      1.196429
2002-05-24      1.210000
2002-05-28      1.157143
2002-05-29      1.103571
2002-05-30      1.071429
2002-05-31      1.076429
2002-06-03      1.128571
2002-06-04      1.117857
2002-06-05      1.147143
2002-06-06      1.182143
2002-06-07      1.118571
2002-06-10      1.156429
2002-06-11      1.153571
2002-06-12      1.092857
2002-06-13      1.082857
2002-06-14      0.986429
2002-06-17      0.922143
2002-06-18      0.910714
2002-06-19      0.951429

1 Ответ

0 голосов
/ 29 января 2019

Хорошо, вот что у меня есть.Не элегантно, но я думаю, что это делает работу. 

import datetime
import pandas as pd
import random

# Sample dataset
df = pd.DataFrame({'Date':sorted([datetime.date.today() - datetime.timedelta(days=i) for i in range(80)]), 'Price':[i for i in range(80)]})
df['Price'] = df['Price'].apply(lambda x: random.uniform(80,100))

>>>df
          Date      Price
0   2018-11-11  83.894664
1   2018-11-12  86.472656
2   2018-11-13  92.566676
3   2018-11-14  94.145586
..         ...        ...
77  2019-01-27  86.338076
78  2019-01-28  95.374173
79  2019-01-29  98.829077

[80 rows x 2 columns]

# MA calculation - borrowing from @jezrael's solution at /10475771/kak-mne-vypolnit-skolzyaschuy-srednyy-v-pande-s-kolonkoi-kotoraya-dolzhna-byt-unikalnoi#10475776
seq = df.set_index('Date')['Price'].rolling(50).mean().rename('MA')
df = df.join(seq, on='Date')

# Set buy signals if current price is higher than 50-day MA
df['Buy'] = df['Price'] - df['MA'] > 0
# Lookbehind check to see that it was below MA at T-1
df['BelowMA'] = df['Price'] - df['MA'] <= 0
# Profit calculations
df['Profit'] = (df['Price'] - df['Price'].shift(-3))/df['Price']

И, наконец, строки, которые удовлетворяют заданным ограничениям 

>>> df.loc[(df['BelowMA'].shift(1)==True) & (df['Buy'] == True)]
          Date      Price         MA   Buy    Profit  BelowMA
54  2019-01-04  91.356371  89.634529  True  0.000697    False
56  2019-01-06  94.116616  89.623909  True  0.114375    False
60  2019-01-10  89.383369  89.254519  True -0.082870    False
63  2019-01-13  96.790575  88.977660  True  0.029314    False
69  2019-01-19  92.193939  89.052057  True  0.071259    False
74  2019-01-24  91.392465  89.302293  True  0.020673    False
76  2019-01-26  95.369195  89.292715  True  0.158250    False
...