U Kv fE@s"ddlZddlmZd ddZdS) N)utilsFTbothcC s" ddlm}t|\}}t|}dd}||}||}||}||}t|}|dkrft|}|dkrxt|}|dkrt|}d}tt |}|dk rtt |}|dkrd}d}n*|dkrd}|}n|dk}|dkr|nd}||}| d | d |dkr&t t |}| n|}| dkrJt t |}| n| }i}t|D]n}|dk rz|||krzqZ| dk r||| krqZ||||f} | |krg|| <|| |qZ||j}!|!j|!j}"}#t|}$d \}%}&| r||j|j}'n||j|j}'||#}(| rL|&|%||(t|$})n|&|%t|$})| dk rl| nt t | | rddd t}*|*dkr|)d t|*d}+nd }+fddD},dddddddg}-|dkrddD}t|*D]v}.|.}/d||/kr6|-|.t|-||/d<d||/krPd||/d<d||/kr|+dkrldnd||/d<q|dkrddD}t|*D]P}.|.}/d||/krd||/d<d ||/kr|+dkrd!nd"||/d <q| r|dkr |&|)d!n|&}0n |%|)d!}0t }1g}2|D]ȉ|rH| r|dkrV|0|)d!n|0}3|jd |3|3f|0d#|(|0d#|(fd$|jdd%|jd|0d&|(d'd'|jd(}4|4d)|0|(8}0ntfd*d+|D}5|j|0|)d!d,|0|5d|)|)d!d,fd-d.d$|jddd/|j|0|5d|)d!d0d'd1|jd(}4|4d)d}6|D]}7|7f|krhqP| r~|j|0dd2n|j|0dd2|dk r|7|}8t|8d!kr|8d|8d}9}:n |7d}9}:n |7d}9}:|dk r||9}9|dk r||:}:| r|6d!dkr| rX|jd |0|)d!|0|)d!f|0|)d!|0|)d!fd3|jdd%n,|j|0|)d!|0|)d!fd4d5d3|jdd%|6d7}6| d6kr| r|jd7|"|0|9d8d'|jd9d:n|j|0d7|#|9d'd;|jd9d:| dd'|jd9d:n"|j|0dd=|#|:d'd1|jd9d:|2|0t!||7fD]v\};}|dk rVt"||<r|>||<|>||<}?}@n"|>||<d|>||<d}?}@| r$|j#|?|@g|0|=|0|=gd@fdA|'i|||<n2|j#|0|=|0|=g|?|@gd@fdA|'i|||<||<}A|A|1krl|And}B|1$|A| r|j#|>g|0|=gfdB|'|BdC||An(|j#|0|=g|>gfdB|'|BdC||Aq\| r|0|)8}0n|0|)7}0qPq*| r|j%&d1|j'&dD|(g|j)d>*dD|j)d8*dD|j)d;*dD|j)d1+dEd7|#f|,dd|j'-|2|j.dddFn|j'&d>|j%&dD|/g|j)d1*dD|j)d8*dD|j)d;*dD|j)d>+dEd7|"f|0dd|j%-|2|j.dddG|S)Ha- Dot plotting (also known as forest and blobbogram). Produce a dotplot similar in style to those in Cleveland's "Visualizing Data" book ([1]_). 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 : str 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 : bool 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 horizontal is False). If None (default), the order is lexical. striped : bool If True, every other line is enclosed in a shaded box. horizontal : bool If True (default), the lines are drawn horizontally, otherwise they are drawn vertically. show_names : str 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 : callable The left/top margin names are passed through this function before drawing on the plot. fmt_right_name : callable 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. References ---------- .. [1] Cleveland, William S. (1993). "Visualizing Data". Hobart Press. .. [2] Jacoby, William G. (2006) "The Dot Plot: A Graphical Display for Labeled Quantitative Values." The Political Methodologist 14(1): 6-14. 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) rNcSs|dkr dSt|S)N)npasarray)xrA/tmp/pip-unpacked-wheel-2v6byqio/statsmodels/graphics/dotplots.pyzdot_plot..g?FTrg{Gz?)rrg@gcsi|]}||qSr)index.0r) style_codesrr szdot_plot..rgbykZpurpleZorangecSsi|] }|iqSrrrrrr rscolormarkeroms cSsi|] }|iqSrrrrrr rsZgreyZ linewidthgffffff?Zdarkgrey)r transformzordergffffff?center)horizontalalignmentverticalalignmentrZboldcsg|]}|dkr|qS)rr)rr)k0rr =s zdot_plot..g{Gz?))\(?r&)(\?r')rrr Zclip_ongRQ?bottom)rZ lightgrey)rr)rr)leftrgrightZ monospace)r"r#rfamilytop)r*rg?r)-rNone)ZlsrlabelnoneZaxes)Zscaleytight)Zscalexr2)1Zmatplotlib.transforms transformsrZ create_mpl_axrrlenZarangezerossetZ set_xmarginZ set_ymarginlistsortrangeappendZget_window_extentZ transformedZdpi_scale_transinvertedwidthheightZblended_transform_factoryZ transDataZ transAxesfloatZ fill_betweentextZset_fontweightZaxhlineZaxvlinesplitlower enumerateZisscalarZplotaddZxaxisZset_ticks_positionZyaxisZset_yticklabelsZspinesZ set_colorZ set_positionZset_ylimZ set_ticksZautoscale_viewZset_xticklabelsZset_xlim)CZpointsZ intervalslinessectionsZstylesZ marker_propsZ line_propsZ split_namesZ section_orderZ line_orderZstackedZ styles_orderZstripedZ horizontalZ show_namesZ fmt_left_nameZfmt_right_nameZshow_section_titlesZaxr3ZfigZasarray_or_noneZnpointZsection_title_spaceZnsectZdraw_section_titlesZ nsect_titleZsection_space_totalZlines0Zlines1Z lines_mapikyZbboxZawidthZaheightZnrowsr(r,ZtransZtitle_space_axesZdposnvalZstackdZstyle_codes_mapcolorsjZscposZlabeledZticksZy0txtmZjrowZk1usZ left_labelZ right_labelZjiZjpZyoptZlcbZucbslZsllr)r$rr dot_plotsw                                                             rQ)NNNNNNNNNFNFTrNNNN)ZnumpyrrrQrrrr s$