29 lines
1007 B
Python
29 lines
1007 B
Python
|
import pandas as pd
|
||
|
|
from pandas_datareader import data
|
||
|
|
|
||
|
|
start_date = '2014-01-01'
|
||
|
|
end_date = '2018-01-01'
|
||
|
|
SRC_DATA_FILENAME = 'goog_data.pkl'
|
||
|
|
|
||
|
|
try:
|
||
|
|
goog_data2 = pd.read_pickle(SRC_DATA_FILENAME)
|
||
|
|
except FileNotFoundError:
|
||
|
|
goog_data2 = data.DataReader('GOOG', 'yahoo', start_date, end_date)
|
||
|
|
goog_data2.to_pickle(SRC_DATA_FILENAME)
|
||
|
|
|
||
|
|
goog_data = goog_data2.tail(620)
|
||
|
|
lows = goog_data['Low']
|
||
|
|
highs = goog_data['High']
|
||
|
|
print(goog_data)
|
||
|
|
|
||
|
|
import matplotlib.pyplot as plt
|
||
|
|
fig = plt.figure()
|
||
|
|
ax1 = fig.add_subplot(111, ylabel='Google price in $')
|
||
|
|
highs.plot(ax=ax1, color='c', lw=2.)
|
||
|
|
lows.plot(ax=ax1, color='y', lw=2.)
|
||
|
|
|
||
|
|
# Return the first 200 rows using pandas.DataFrame.head(200)
|
||
|
|
plt.hlines(highs.head(200).max(), lows.index.values[0], lows.index.values[-1], linewidth=2, color='g')
|
||
|
|
plt.hlines(lows.head(200).min(), lows.index.values[0], lows.index.values[-1], linewidth=2, color='r')
|
||
|
|
plt.axvline(linewidth=2, color='b', x=lows.index.values[200], linestyle=':')
|
||
|
|
plt.savefig("supportresistanceline.png")
|