From a5ce8ff53f87bc3724b33499166551c76f6e25fc Mon Sep 17 00:00:00 2001 From: Jens Ahrensfeld Date: Fri, 13 Dec 2019 19:10:05 +0000 Subject: [PATCH] improved git-svn-id: http://moon:8086/svn/projects/Stock@304 fda53097-d464-4ada-af97-ba876c37ca34 --- pandas_eval.py | 31 ++++++++++++++++--------------- 1 file changed, 16 insertions(+), 15 deletions(-) diff --git a/pandas_eval.py b/pandas_eval.py index 9cca267..74cc0f3 100644 --- a/pandas_eval.py +++ b/pandas_eval.py @@ -103,8 +103,9 @@ titles.append('QCOM') titles.append('DPW.DE') titles.append('CSCO') titles.append('AIR') +titles.append('NVDA') -show_title = 'QCOM' +show_title = 'XLNX' show_range_days = 20 pd.DatetimeIndex @@ -139,6 +140,17 @@ def fetch(title): data = hdf[title] hdf.close() +def plot(ax, fig, data): + dates = data.index + data.plot() + def hover(event): + if event.inaxes == ax: + x_idx = int(event.xdata) + ax.format_xdata = lambda x: dates[x_idx] + ax.format_ydata = lambda y: '{:.2f}'.format(data[x_idx]) + + fig.canvas.mpl_connect("motion_notify_event", hover) + def show(title): k_euro = 1/1.11 @@ -156,23 +168,12 @@ def show(title): data = data_full[len(data_full)-1 - show_range_days:] boll = boll_full[len(boll_full)-1 - show_range_days:] - dates = data.index fig = plt.figure(1) - ax = plt.subplot(311) - - def hover(event): - if event.inaxes == ax: - x_idx = int(event.xdata) - ax.format_xdata = lambda x: dates[x_idx] - ax.format_ydata = lambda y: '{:.2f}'.format(data['close'][x_idx]) - - fig.canvas.mpl_connect("motion_notify_event", hover) - - data['close'].plot() - data['ema'].plot() - data['sma'].plot() + plot(ax, fig, data['close']) + plot(ax, fig, data['ema']) + plot(ax, fig, data['sma']) plt.title(title) plt.legend() plt.grid()