Professional Documents
Culture Documents
Untitled
Untitled
import pandas as pd
api_key = 'OQ9JRGVKDLDSH096'
base_url = 'https://www.alphavantage.co/query'
params = {
'function': 'TIME_SERIES_INTRADAY',
'symbol': symbol,
'interval': interval,
'apikey': api_key
data = data.astype(float)
data = data.sort_index()
return data
data['sma20'] = data['close'].rolling(window=20).mean()
data['sma50'] = data['close'].rolling(window=50).mean()
data['sma200'] = data['close'].rolling(window=200).mean()
data['position'] = 0
for i in range(len(data)):
if data['buy_signal'][i]:
data['position'][i] = 1
elif data['sell_signal'][i]:
data['position'][i] = -1
data['cumulative_return'] = (1 + data['daily_return']).cumprod()
return data
# print result
print(result)