import pandas as pd
[docs]def moving_average(stock_data, size):
"""
Calculates the moving average of a stock price for a given range of dates
Parameters
----------
stock_data : string
A string corresponding to the name of the stock.
size : int
The size of the period that we're interested in.
Returns
--------
moving_avg : dataframe
A dataframe representing the 200 day moving average, for example, of a stock price.
Examples
--------
>>> moving_average("MSFT", 200)
"""
data = pd.read_csv('../../data/'+stock_data+'.csv')
data.index = pd.to_datetime(data["Date"], utc=True).dt.date
mov_avg = {}
mov_avg[f"{size}MA"] = data["Close"].rolling(window=size).mean()
moving_avg = pd.DataFrame(mov_avg).reset_index()
return moving_avg