-
Notifications
You must be signed in to change notification settings - Fork 0
/
code2_tickergraph.py
53 lines (37 loc) · 1.22 KB
/
code2_tickergraph.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Tue Dec 10 20:13:03 2019
@author: itamar
"""
import datetime as dt
import matplotlib.pyplot as plt
from matplotlib import style
from mpl_finance import candlestick_ohlc
import matplotlib.dates as mdates
import pandas as pd
import pandas_datareader.data as web
style.use('ggplot')
start = dt.datetime(2000,1,1)
end = dt.datetime(2019,12,22)
df = web.DataReader('JHSF3.SA','yahoo',start,end)
#df.to_csv('tsla.csv')
#df = pd.read_csv('tsla.csv', parse_dates = True,index_col=0)
df_ohlc = df['Adj Close'].resample('10D').ohlc()
df_volume = df['Volume'].resample('10D').sum()
df_ohlc.reset_index(inplace =True)
df_ohlc['Date'] = df_ohlc['Date'].map(mdates.date2num)
#df['Adj Close'].plot()
#plt.show()
#df['100ma'] = df['Adj Close'].rolling(window = 100, min_periods = 0).mean()
ax1 = plt.subplot2grid((6,1),(0,0), rowspan=5, colspan=1)
ax2 = plt.subplot2grid((6,1),(5,0), rowspan=1, colspan=1,sharex = ax1)
ax1.xaxis_date()
candlestick_ohlc(ax1,df_ohlc.values,width = 2, colorup = 'g')
ax2.fill_between(df_volume.index.map(mdates.date2num),df_volume.values, 0)
"""
ax1.plot(df.index,df['Adj Close'])
ax1.plot(df.index,df['100ma'])
ax2.bar(df.index,df['Volume'])
"""
plt.show()