�
�j�UV �
@ s- d Z d d l j Z d d l Z d d � Z d d � Z d( d d d � Z d d d d
d � Z d d
� Z
d d d d d d � Z d d d � Z d d d d d d � Z
d d d d d d � Z d d � Z d d d! � Z e d" k r)d# d$ d% d& d' d# d$ d% d& d' g
d Z e e � e e � n d S))z6
Created on Fri May 2 12:21:40 2014
@author: rlabbe
� Nc C s� d } x[ | D]S } t j | d g | g d | d g d d d | d d d d
d �| d 7} q
Wt j d d � t j | d | d � t � t j � j j j g � t j � d S)
Ng �?r �xerr� �fmt�o�label� �capthick�capsize�
g�������?)
�plt�errorbar�ylim�xlim� show_legend�gca�axes�yaxis� set_ticks�show)�bars�xlims�i�bar� r �./code/book_plots.py�plot_errorbars
s
Cr c C s t j d d d d � d S)N�locz center left�bbox_to_anchorr � �?)r r )r �legendr r r r r s r r c C s� t j � t j � } t j t | � � } | j | | d d �| rW t j | � n t j | d | � | d k r� t j | � n d S)N�colorz#30a2dag�������?)
r �clar �np�arange�lenr r
�xticks�title)�posr
r& �ax�xr r r �bar_plot s
r* c C sX | d k r t j | � n | d k r8 t j | � n | d k rT t j | � n d S)zL helps make code in book shorter. Optional set title, xlabel and ylabel
N)r �xlabel�ylabelr& )r& r) �yr r r �
set_labels* s r. c C s* t j � j | � t j � j | � d S)z\ helper function to make code in book shorter. Set the limits for the x
and y axis.
N)r r �set_xlim�set_ylim)r) r- r r r �
set_limits5 s r1 �rr Z Measurementsc K sv t j d d � | d k rJ t j | | d | d | d d d | | �n( t j | d | d | d d d | | �d S) zW Helper function to give a consistant way to display
measurements in the book.
�tightTNr �lw�lsz--r )r � autoscale�plot)�xs�ysr r4 r �kwargsr r r �plot_measurements> s .r; g �?c C s� t j | � | } t j | d d d d d d �t j | d d d d d d �t j t t | � � | | d d d d
�d S) z� plots standand deviation given in Ps as a yellow shaded region. One std
by default, use stds for a different choice (e.g. stds=3 for 3 standard
deviations.
r �kr5 �:r4 r � facecolorz#ffff00�alphag333333�?N)r" �sqrtr r7 � fill_between�ranger$ )ZPsZstds�stdr r r �plot_residual_limitsV s
rD ZTrackr< c K sf | d k r: t j | | d | d | d d d | | �n( t j | d | d | d d d | | �d S)Nr r4 r5 r= r )r r7 )r8 r9 r �cr4 r: r r r �
plot_trackd s .rF z#013afe�Filterc K s� | d k r' | } t t | � � } n t j | | d | d | | �| d k rV d St j | � } t j | � } | | } | | } t j | | | d d d d d d �t j | | | d d d d d d �t j | | | d d d
d �d S) Nr r � linestyler= r< r4 r r> �yellowr? g�������?)rB r$ r r7 r" �asarrayr@ rA ) r8 r9 rE r �varsr: rC Zstd_topZstd_btmr r r � plot_filterk s
&&rL c C s� t j | � d } t j | | | | | | | | g � } t j | | | | | | | | g � } t j | | | d | �d S)z\
Draws a square-shaped blob with the given area (< 1) at
the given coordinates.
r � edgecolorN)r" r@ �arrayr �fill)r) r- �areaZcolour�hsZxcornersZycornersr r r �_blob� s ++rR c
C s� d } t j � r t j � n t j � | j \ } } | s| d t j t j t j t j | � � � t j d � � } n t j
t j d | | d g � t j d d | | g � d � t j d � t j d � x� t
| � D]� } x� t
| � D]� } | d } | d } | | | f } | d k rXt | d | | d t d | | � d � q� | d k r� t | d | | d t d | | � d
� q� q� Wq� W| r�t j � n d S) z�
Draws a Hinton diagram for visualizing a weight matrix.
Temporarily disables matplotlib interactive mode if it is on,
otherwise this takes forever.
Fr r �gray�off�equalr g �?�white�blackN)r �
isinteractive�ioff�clf�shaper" �ceil�log�max�absrO rN �axisrB rR �min�ion)
�WZ maxweightZreenable�height�widthr) r- �_x�_y�wr r r �hinton� s:
>
ri �__main__g�>�$E��?g�L����?g ���\G�?g�,�Wc��?g1�I�+�?)r r )�__doc__�matplotlib.pyplot�pyplotr �numpyr" r r r* r. r1 r; rD rF rL rR ri �__name__�pr r r r �<module> s$
(
