p7]c@soddlZddlmZddlmZdddddddddedeeddddddZ dS(iN(trangei(tutilstbothcEs ddlj}tj|\}}tj|}d}||}||}||}||}t|}|dkrtj|}n|dkrtj |}n|dkrtj |}nd}tt |}|dk r tt |}n|t kr&t }d}n?|t krAt }|}n$|dk}|dkr_|nd}||}|j d|jd|dkrtt |}|jn|}| dkrtt |}|jn| }i}xt|D]}|dk r&|||kr&qn| dk rH||| krHqn||||f} | |krug|| |jd0|"|0|;d#d1d%d$d |jd2d3qs |j|0d0|#|;d#d$d%d4d |jd2d3n|j#dHkr |<dk r | r |jdd5|"|0|<d#d.d%d$d |jd2d3q |j|0dd5|#|<d#d$d%d,d |jd2d3q n|2j|0xt$||3|9fD]\}=}>d}?| rf |) d6|,||>|+}?n||>}@|dk rP tj%||>r |@||>|@||>}A}Bn%|@||>d|@||>d}A}B| r |j&|A|Bg|0|?|0|?gd7d |'|||>qP |j&|0|?|0|?g|A|Bgd7d |'|||>n||>}C|C|1krl |Cnd}D|1j'|C| r |j&|@g|0|?gd8d9d |'d:|D||Cq0 |j&|0|?g|@gd8d9d |'d:|D||Cq0 W| r |0|)8}0qN|0|)7}0qNWqW| r |j(j)d,|j*j)d;|j+g|j,d.j-d;|j,d1j-d;|j,d4j-d;|j,d,j.d<d0|#f|j/dd|j*j0|2|j1d=t d>t n|j*j)d.|j(j)d;|j2g|j,d,j-d;|j,d1j-d;|j,d4j-d;|j,d.j.d<d0|"f|j3dd|j(j0|2|j1d?t d>t |S(Is Produce a dotplot similar in style to those in Cleveland's "Visualizing Data" book. These are also known as "forest plots". Parameters ---------- points : array_like The quantitative values to be plotted as markers. intervals : array_like The intervals to be plotted around the points. The elements of `intervals` are either scalars or sequences of length 2. A scalar indicates the half width of a symmetric interval. A sequence of length 2 contains the left and right half-widths (respectively) of a nonsymmetric interval. If None, no intervals are drawn. lines : array_like A grouping variable indicating which points/intervals are drawn on a common line. If None, each point/interval appears on its own line. sections : array_like A grouping variable indicating which lines are grouped into sections. If None, everything is drawn in a single section. styles : array_like A grouping label defining the plotting style of the markers and intervals. marker_props : dict A dictionary mapping style codes (the values in `styles`) to dictionaries defining key/value pairs to be passed as keyword arguments to `plot` when plotting markers. Useful keyword arguments are "color", "marker", and "ms" (marker size). line_props : dict A dictionary mapping style codes (the values in `styles`) to dictionaries defining key/value pairs to be passed as keyword arguments to `plot` when plotting interval lines. Useful keyword arguments are "color", "linestyle", "solid_capstyle", and "linewidth". split_names : string If not None, this is used to split the values of `lines` into substrings that are drawn in the left and right margins, respectively. If None, the values of `lines` are drawn in the left margin. section_order : array_like The section labels in the order in which they appear in the dotplot. line_order : array_like The line labels in the order in which they appear in the dotplot. stacked : boolean If True, when multiple points or intervals are drawn on the same line, they are offset from each other. styles_order : array_like If stacked=True, this is the order in which the point styles on a given line are drawn from top to bottom (if horizontal is True) or from left to right (if horiontal is False). If None (default), the order is lexical. striped : boolean If True, every other line is enclosed in a shaded box. horizontal : boolean If True (default), the lines are drawn horizontally, otherwise they are drawn vertically. show_names : string Determines whether labels (names) are shown in the left and/or right margins (top/bottom margins if `horizontal` is True). If `both`, labels are drawn in both margins, if 'left', labels are drawn in the left or top margin. If `right`, labels are drawn in the right or bottom margin. fmt_left_name : function The left/top margin names are passed through this function before drawing on the plot. fmt_right_name : function The right/bottom marginnames are passed through this function before drawing on the plot. show_section_titles : bool or None If None, section titles are drawn only if there is more than one section. If False/True, section titles are never/always drawn, respectively. ax : matplotlib.axes The axes on which the dotplot is drawn. If None, a new axes is created. Returns ------- fig : Figure The figure given by `ax.figure` or a new instance. Notes ----- `points`, `intervals`, `lines`, `sections`, `styles` must all have the same length whenever present. Examples -------- This is a simple dotplot with one point per line: >>> dot_plot(points=point_values) This dotplot has labels on the lines (if elements in `label_values` are repeated, the corresponding points appear on the same line): >>> dot_plot(points=point_values, lines=label_values) References ---------- * Cleveland, William S. (1993). "Visualizing Data". Hobart Press. * Jacoby, William G. (2006) "The Dot Plot: A Graphical Display for Labeled Quantitative Values." The Political Methodologist 14(1): 6-14. iNcSs|dkrdStj|S(N(tNonetnptasarray(tx((s<lib/python2.7/site-packages/statsmodels/graphics/dotplots.pyttg?iig{Gz?g@gcs"i|]}j||qS((tindex(t.0R(t style_codes(s<lib/python2.7/site-packages/statsmodels/graphics/dotplots.pys s trtgtbtytktpurpletorangecSsi|]}i|qS(((R R((s<lib/python2.7/site-packages/statsmodels/graphics/dotplots.pys s tcolortmarkertotmsi icSsi|]}i|qS(((R R((s<lib/python2.7/site-packages/statsmodels/graphics/dotplots.pys s tgreyt linewidthiigffffff?tdarkgreyt transformtzordergffffff?thorizontalalignmenttcentertverticalalignmenttboldg{Gz?g)\(?g(\?tclip_ongRQ?tbottomt lightgreytleftRgtrighttfamilyt monospacettopg?it-tlsRtlabeltnonetaxestscaleyttighttscalex(ii(ii(g)\(?g)\(?(g(\?g(\?(ii(ii(ii(R#R(R$R(4tmatplotlib.transformst transformsRt create_mpl_axRRtlenRtarangetzerostsettFalsetTruet set_xmargint set_ymargintlisttsortRtappendtget_window_extentt transformedtdpi_scale_transtinvertedtwidththeighttblended_transform_factoryt transDatat transAxestfloatt fill_betweenttexttset_fontweighttaxhlinetaxvlinetsplittlowert enumeratetisscalartplottaddtxaxistset_ticks_positiontyaxistset_yticklabelstspinest set_colort set_positiontset_ylimt set_tickstautoscale_viewtset_xticklabelstset_xlim(Etpointst intervalstlinestsectionststylest marker_propst line_propst split_namest section_ordert line_ordertstackedt styles_ordertstripedt horizontalt show_namest fmt_left_nametfmt_right_nametshow_section_titlestaxR1tfigtasarray_or_nonetnpointtsection_title_spacetnsecttdraw_section_titlest nsect_titletsection_space_totaltlines0tlines1t lines_maptitkytbboxtawidthtaheighttnrowsR!R'ttransttitle_space_axestdpostnvaltstackdtstyle_codes_maptcolorstjtsctpostlabeledttickstk0ty0ttxtRtmtjrowtk1tust left_labelt right_labeltjitjptyotpttlcbtucbtsltsll((R s<lib/python2.7/site-packages/statsmodels/graphics/dotplots.pytdot_plotst                                      '  '#  $     /   "      "                #   &&       ( tnumpyRtstatsmodels.compatRRRRR7R8R(((s<lib/python2.7/site-packages/statsmodels/graphics/dotplots.pyts