Ñò
㖪Jc @ s; d Z d d k Z d d k Z d d k Z d d k l Z y d d k l Z Wn# e j
o d d k l Z n Xy d d k Z Wn e j
o e d ƒ ‚ n Xd d k
Z d d k l
Z
l Z d d k l Z d d k l Z d d k l Z l Z d d
k l Z d d k l Z l Z l Z d d k l Z d d
k l Z e ƒ Z e i! i" e d ƒ Z# e$ e% d „ Z& d d „ Z( e$ e% d d „ Z) d d d d „ Z* d d d d d „ Z+ d d d d „ Z, d d d d d „ Z- d d d d d „ Z. d d d d d „ Z/ d d d d d „ Z0 d d! d d d" „ Z1 d S(# so
A collection of modules for collecting, analyzing and plotting
financial data. User contributions welcome!
iÿÿÿÿN( t urlopen( t md5s8 The finance module requires datetime support (python2.3)( t verboset
get_configdir( t date2num( t Bunch( t LineCollectiont PolyCollection( t colorConverter( t Line2Dt TICKLEFTt TICKRIGHT( t Rectangle( t Affine2Ds
finance.cachec C s g } | i ƒ } d } xx| d D]l} | i d ƒ } t | ƒ d j o q# n | d } | d j oI y) d } t i t i | | ƒ d Œ } Wq² t j
o
d } q² Xn t i t i | | ƒ d Œ } t | ƒ }
g } | d d !D] } | t
| ƒ qï ~ \ }
} } } t | d ƒ } | oB t
| d ƒ } | | } |
| 9}
| | 9} | | 9} | } n | i |
|
| | | | f ƒ q# W| i
ƒ | on t | ƒ d j o d St t i t | Œ ƒ \ } }
} } } } t d
| d |
d | d
| d | d | ƒ S| Sd S( s
Parse the historical data in file handle fh from yahoo finance and return
results as a list of
d, open, close, high, low, volume
where d is a floating poing representation of date, as returned by date2num
if adjust=True, use adjusted prices
i t ,i i s %Y-%m-%di s %d-%b-%yi i t datet opent closet hight lowt volumeN( t readlinest Nonet splitt lent datetimeR t timet strptimet
ValueErrorR t floatt intt appendt reverset mapt npt asarrayt zipR ( t fht asobjectt adjustedt resultst linest datefmtt linet valst datestrt dtt dt _[1]t valR R R R R t acloset mR ( ( s8 /usr/lib64/python2.6/site-packages/matplotlib/finance.pyt parse_yahoo_historical$ sD
# :
#
*,c
C ss | i ƒ } | i d | i | i f } | i d | i | i f } d } | | d | d | d | d | d | d | f } | d j o% t i i t t | ƒ i
ƒ ƒ } n t i i | ƒ o' t | ƒ } t
i d | | f ƒ nƒ t i i t ƒ p t i t ƒ n t | ƒ } t | d ƒ } | i | i ƒ ƒ | i ƒ t
i d | | f ƒ t | d ƒ } | S(
se
Fetch historical data for ticker between date1 and date2. date1 and
date2 are datetime instances
Ex:
fh = fetch_historical_yahoo('^GSPC', d1, d2)
cachename is the name of the local file cache. If None, will
default to the md5 hash or the url (which incorporates the ticker
and date range)
a file handle is returned
i s_ http://table.finance.yahoo.com/table.csv?a=%d&b=%d&c=%d&d=%d&e=%d&f=%d&s=%s&y=0&g=d&ignore=.csvi i s Using cachefile %s for %st ws Saved %s data to cache file %st rN( t uppert montht dayt yearR t ost patht joint cachedirR t hexdigestt existst fileR t reportt isdirt mkdirR t writet readR (
t tickert date1t date2t cachenamet d1t d2t urlFmtt urlR% t urlfh( ( s8 /usr/lib64/python2.6/site-packages/matplotlib/finance.pyt fetch_historical_yahooX s(
%
c C sh t | | | | ƒ } y t | | | ƒ } Wn6 t j
o* } t i d t d | i d ƒ d SX| S( sŒ
Get historical data for ticker between date1 and date2. date1 and
date2 are datetime instances
results are a list of tuples
(d, open, close, high, low, volume)
where d is a floating poing representation of date, as returned by date2num
if asobject is True, the return val is an object with attrs date,
open, close, high, low, volume, which are equal length arrays
if adjust=True, use adjusted prices
Ex:
sp = f.quotes_historical_yahoo('^GSPC', d1, d2, asobject=True, adjusted=True)
returns = (sp.open[1:] - sp.open[:-1])/sp.open[1:]
[n,bins,patches] = hist(returns, 100)
mu = mean(returns)
sigma = std(returns)
x = normpdf(bins, mu, sigma)
plot(bins, x, color='red', lw=2)
cachename is the name of the local file cache. If None, will
default to the md5 hash or the url (which incorporates the ticker
and date range)
s urlopen() failure
s
i N( RP R4 t IOErrort warningst warnRN t strerrorR ( RG RH RI R& R' RJ R% t rett exc( ( s8 /usr/lib64/python2.6/site-packages/matplotlib/finance.pyt quotes_historical_yahoo‡ s i t kR6 c C s7 g } x | D]} | d \ } } } }
} | | j o
| } n | } t d | | f d | |
f d | d t ƒ }
t d | | f d | | f d | d t d t d | ƒ } t d | | f d | | f d | d t d | d t ƒ } | i |
| | f ƒ | i |
ƒ | i | ƒ | i | ƒ q
W| i ƒ | S( s%
quotes is a list of (time, open, close, high, low, ...) tuples
Represent the time, open, close, high, low as a vertical line
ranging from low to high. The left tick is the open and the right
tick is the close.
time must be in float date format - see date2num
ax : an Axes instance to plot to
ticksize : open/close tick marker in points
colorup : the color of the lines where close >= open
colordown : the color of the lines where close < open
return value is a list of lines added
i t xdatat ydatat colort antialiasedt markert
markersize( R t FalseR
R t extendt add_linet autoscale_view( t axt quotest ticksizet colorupt colordownR) t qt tR R R R R[ t vlinet olinet cline( ( s8 /usr/lib64/python2.6/site-packages/matplotlib/finance.pyt plot_day_summary® s:
gš™™™™™É?g ð?c
C s6 | d } g } g } x | D]} | d \ }
} } }
} | | j o | } | } | | } n | } | } | | } t d |
|
f d | |
f d d d d d t ƒ } t d
|
| | f d | d | d
| d | ƒ } | i | ƒ | i | ƒ | i | ƒ | i | ƒ | i | ƒ q W| i ƒ | | f S( s‰
quotes is a list of (time, open, close, high, low, ...) tuples.
As long as the first 5 elements of the tuples are these values,
the tuple can be as long as you want (eg it may store volume).
time must be in float days format - see date2num
Plot the time, open, close, high, low as a vertical line ranging
from low to high. Use a rectangular bar to represent the
open-close span. If close >= open, use colorup to color the bar,
otherwise use colordown
ax : an Axes instance to plot to
width : fraction of a day for the rectangle width
colorup : the color of the rectangle where close >= open
colordown : the color of the rectangle where close < open
alpha : the rectangle alpha level
return value is lines, patches where lines is a list of lines
added and patches is a list of the rectangle patches added
g @i RY RZ R[ RX t linewidthg à?R\ t xyt widtht heightt facecolort edgecolor( R t TrueR t set_alphaR Ra t add_patchRb ( Rc Rd Rp Rf Rg t alphat OFFSETR) t patchesRh Ri R R R R R[ t lowerRq Rj t rect( ( s8 /usr/lib64/python2.6/site-packages/matplotlib/finance.pyt candlestickì s>
i c ) C s£ g } t t t | ƒ ƒ | | ƒ D]9 \ } }
} |
d j o | | |
f | | f f q# q# ~ } | d f d f g }
d
| d f f g } g } t t t | ƒ ƒ | ƒ D]* \ } } | d j o | | | f q° q° ~ } g } t t t | ƒ ƒ | ƒ D]* \ } } | d j o | | | f qq~ } | i i d d } t ƒ i | d ƒ } t i | ƒ \ } } } | | | d f } t i | ƒ \ } } } | | | d f } h | t 6| t
6} g } t | | ƒ D]; \ } } | d j o" | d j o | | | | j qÛqÛ~ } d } d } t | d | d | d | ƒ} t |
d | d | d | d
| d | i ƒ} | i
| ƒ t | d | d | d | d
| d | i ƒ}! |! i
| ƒ d t | ƒ }" }# t g }$ | D] }
|
d j o |$ |
qÞqÞ~$ ƒ }% t g }& | D] } | d j o |& | qq~& ƒ }' |" |% f |# |' f f }( | i |( ƒ | i ƒ | i | ƒ | i | ƒ | i |! ƒ | | |! f S( s³
Represent the time, open, close, high, low as a vertical line
ranging from low to high. The left tick is the open and the right
tick is the close.
ax : an Axes instance to plot to
ticksize : size of open and close ticks in points
colorup : the color of the lines where close >= open
colordown : the color of the lines where close < open
return value is a list of lines added
iÿÿÿÿi g ð?g R@g i t colorst
linewidthst antialiasedst offsetst transOffset( i i ( i i ( i ( i ( R$ t xrangeR t figuret dpiR
t scaleR t to_rgbRt R_ R t transDatat
set_transformt mint maxt update_datalimRb t add_collection() Rc t openst closest highst lowsRe Rf Rg R0 t iR R t
rangeSegmentst openSegmentst
closeSegmentst _[2]R t offsetsOpent _[3]R t offsetsCloseR… t
tickTransformR6 t gt bt colordt _[4]R} t useAAt lwt rangeCollectiont openCollectiont closeCollectiont minpyt maxxt _[5]t minyt _[6]t maxyt corners( ( s8 /usr/lib64/python2.6/site-packages/matplotlib/finance.pyt plot_day_summary22 sX eSS
X
88
g è?c $ C sç | d } g }
t t t | ƒ ƒ | | ƒ D]h \ } } }
| d j oL |
d j o? |
| | | f | | |
f | | |
f | | | f f q- q- ~
} g } t t t | ƒ ƒ | | ƒ D]9 \ } } } | d j o | | | f | | f f qÁ qÁ ~ } t i | ƒ \ } } } | | | | f } t i | ƒ \ } } } | | | | f } h | t 6| t 6} g } t | | ƒ D]; \ } }
| d j o" |
d j o | | | |
j qq~ } d } d } t | d d d | d | ƒ} t | d | d
d d | d | ƒ} d t | ƒ } } t g } | D] } | d j o | | q2q2~ ƒ } t
g }! | D] } | d j o |! | qjqj~! ƒ }" | | f | |" f f }# | i |# ƒ | i ƒ | i
| ƒ | i
| ƒ | | f S( s‡
Represent the open, close as a bar line and high low range as a
vertical line.
ax : an Axes instance to plot to
width : the bar width in points
colorup : the color of the lines where close >= open
colordown : the color of the lines where close < open
alpha : bar transparency
return value is lineCollection, barCollection
g @iÿÿÿÿi g à?R} i R~ R t
facecolorst
edgecolors( i ( g à?( i i i i ( ( i i i i ( i i i i ( ( i i i i ( R$ R‚ R R R† Rt R_ R R R‰ RŠ R‹ Rb RŒ ($ Rc R RŽ R R Rp Rf Rg Rw t deltaR0 R‘ R R t barVertsR• R R R’ R6 Rš R› Rœ R— R} Rž RŸ R t
barCollectiont minxR¤ R R¦ R¥ R¨ R© ( ( s8 /usr/lib64/python2.6/site-packages/matplotlib/finance.pyt candlestick2 s<
”e
X 88
c C s™ t i | ƒ \ } } }
| | |
| f } t i | ƒ \ } } }
| | |
| f } h | t 6| t 6} g } t | | ƒ D]; \ }
} |
d j o" | d j o | | |
| j q| q| ~ } | d } g } t | ƒ D]X \ } } | d j o? | | | d f | | | f | | | f | | d f f qÛ qÛ ~ } t | d | d d d d d d
ƒ} d t | ƒ t | ƒ f f } | i | ƒ | i
ƒ | S( s½
Add a volume overlay to the current axes. The opens and closes
are used to determine the color of the bar. -1 is missing. If a
value is missing on one it must be missing on all
ax : an Axes instance to plot to
width : the bar width in points
colorup : the color of the lines where close >= open
colordown : the color of the lines where close < open
alpha : bar transparency
iÿÿÿÿg @i R« R¬ i R R~ g à?( i i i i ( ( i i i i ( i ( g à?( i i ( R R† Rt R_ R$ t enumerateR R RŠ R‹ Rb ( Rc R RŽ t volumesRf Rg Rp Rw R6 Rš R› Rœ R0 R R R} R R• R‘ t vt barsR¯ R© ( ( s8 /usr/lib64/python2.6/site-packages/matplotlib/finance.pyt volume_overlayÕ s$
X
r
c C s+ t | | d | d | d | | | | ƒ S( s
Add a volume overlay to the current axes. The closes are used to
determine the color of the bar. -1 is missing. If a value is
missing on one it must be missing on all
ax : an Axes instance to plot to
width : the bar width in points
colorup : the color of the lines where close >= open
colordown : the color of the lines where close < open
alpha : bar transparency
nb: first point is not displayed - it is used only for choosing the
right color
iÿÿÿÿi ( R¶ ( Rc RŽ R³ Rf Rg Rp Rw ( ( s8 /usr/lib64/python2.6/site-packages/matplotlib/finance.pyt volume_overlay2 s c - C sñ t i | ƒ \ } } } | | | | f } t i | ƒ \ } } } | | | | f } h | t 6| t 6} t | Œ \ }
} } }
} } g } t | d | d ƒ D]; \ } } | d j o" | d j o | | | | j q¢ q¢ ~ } | i d | | d | d j ƒ | d } | d } g } | D]C \ } } } } } } | | d f | | f | | f | d f f q(~ } | i i d d } | i i | i
i } t ƒ i | | ƒ } g }! | D] \ } } } } } } |! | q¾~! }
g }" |
D] } |" | d f qñ~" }# d }$ d }% t
| d | d d d
|$ d |% d |# d
| i ƒ}& |& i | ƒ t |
ƒ t |
ƒ }' }( d }) t g }* | D] \ } } } } } } |* | q†~* ƒ }+ |' |) f |( |+ f f }, | i |, ƒ | i |& ƒ | i ƒ |& S( s‡
Add a volume overlay to the current axes. quotes is a list of (d,
open, close, high, low, volume) and close-open is used to
determine the color of the bar
kwarg
width : the bar width in points
colorup : the color of the lines where close1 >= close0
colordown : the color of the lines where close1 < close0
alpha : bar transparency
iÿÿÿÿi i g @g ð?g R@g à?R« R¬ R R~ R€ R ( i ( g à?( i i i i ( ( i i i i ( R R† Rt R_ R$ t insertRƒ R„ t bboxRq t viewLimR
R… R R‡ Rˆ R‰ RŠ R‹ RŒ Rb (- Rc Rd Rf Rg Rp Rw R6 Rš R› Rœ t datesR RŽ R R R³ R0 t close0t close1R} t rightt leftR• R/ R R R R R Rµ t sxt syt barTransformR— R t offsetsBarsRž RŸ R¯ R£ R¤ R¦ R¥ R¨ R© ( ( s8 /usr/lib64/python2.6/site-packages/matplotlib/finance.pyt volume_overlay3 sF
`"
W3'
9
R› t lc C sé t i | | ƒ f } t i | | ƒ f } | d } | d } g }
| D]B } | d j o/ |
| d f | | f | | f | d f f qJ qJ ~
} | i i d d }
| i i | i i } t ƒ i |
| ƒ } g } t | ƒ D]* \ } } | d j o | | d f qå qå ~ } t
| d | d | d d
d d d | d | i ƒ} | i | ƒ d t
| ƒ } } d } t g } | D] } | d j o | | qq~ ƒ } | | f | | f f } | i | ƒ | i ƒ | i | ƒ | S( sÄ
Add a bar collection graph with height vals (-1 is missing).
ax : an Axes instance to plot to
width : the bar width in points
alpha : bar transparency
g @iÿÿÿÿi g ð?g R@R« R¬ R R~ g à?R€ R ( i ( g à?( R t to_rgbaRƒ R„ R¹ Rq Rº R
R… R² R R‡ Rˆ R RŠ R‹ Rb RŒ ( Rc R, Rr Rs Rp Rw R« R¬ R¾ R¿ R0 R´ Rµ RÀ RÁ R R• R‘ Rà R¯ R£ R¤ R¦ R— R¨ R© ( ( s8 /usr/lib64/python2.6/site-packages/matplotlib/finance.pyt index_bar` s2
VD
8
(2 t __doc__R; R RR t urllib2R t hashlibR t ImportErrorR t numpyR" t
matplotlibR R R» R t matplotlib.cbookR t matplotlib.collectionsR R t matplotlib.colorsR R) R R
R Ry R t matplotlib.transformsR
t configdirR< R= R> R_ Rt R4 R RP RW Rm R| Rª R± R¶ R· RÄ RÇ ( ( ( s8 /usr/lib64/python2.6/site-packages/matplotlib/finance.pyt sR $ 4 /'= E\ E+H