M/PhE0ddlZddlmZ ddZdS) N)utilsFTbothc ZCDddlm}tj|\}}t j|}d}||}||}||}||}t |}|t j|}|t j|}|t j|}d}t t|}|t t|}|durd}d}n|durd}|}n|dk}|dkr|nd}||z}| d| d|1tt|}| n|}| 1tt|}| n| }i}t|D]P}| |||vr| ||| vr||||f} | |vrg|| <|| |Q||j}!|!j|!j}#}"t |}$d \}%}&| r!||j|j}'n ||j|j}'||#z }(| r|&|%z ||(zz t3|$z })n|&|%z t3|$z })| | Dn0tt|DD | r Dddd Dt D}*|*dkr|)d t3|*dz zz }+nd }+Dfd DD},gd}-| dDD}t|*D]g}.D|.}/d||/vr!|-|.t |-z||/d<d||/vr d||/d<d||/vr|+dkrdnd||/d<h| dDD}t|*D]<}.D|.}/d||/vr d||/d<d||/vr|+dkrdnd||/d<=| r|dkr|&|)dz z n|&}0n|%|)dz z}0t}1g}2|D]C|r#| rC|dkr|0|)dz zn|0}3|d |3|3f|0d|(zz |0d|(zz fd|jd|d|0d|(zz Cdd|j }4|4d!|0|(z}0nt Cfd"|D}5||0|)dz z d#z|0|5dz |)zz|)dz zd#z fd$d%d|jdd&||0|5dz |)zdz zd'Cdd(|j }4|4d!d}6|D]S}7C|7f|vr | r||0d)n||0d)|>|7|}8t |8dkr|8d|8d}:}9n |7d}:}9n|7d}:}9| ||9}9| ||:}:| rv|6dzdkrm| r=|d |0|)dz z |0|)dz z f|0|)dz z|0|)dz zfd*|jdn.||0|)dz z |0|)dz zfd+d,d*|jd|6dz }6| d-vrK| r%|d.|"z |0|9d/d|jd01n$||0d.|#z |9dd2|jd01| d3vrS|:Q| r(|dd4|"z z|0|:d5d|jd01n'||0dd4|#z z|:dd(|jd01|2|0tC|C|7fD]7\};}|t j"||<r|>||<z |>||<z}@}?n"|>||<dz |>||<dz}@}?| r)|j#|?|@g|0|=z|0|=zgd7fd8|'i|||<n(|j#|0|=z|0|=zg|?|@gd7fd8|'i|||<||<}A|A|1vr|And}B|1$|A| r |j#|>g|0|=zgfd9|'|Bd:||A|j#|0|=zg|>gfd9|'|Bd:||A9| r|0|)z}0N|0|)z }0U| r|j%&d(|j'&d;|(g|j)d5*d;|j)d/*d;|j)d2*d;|j)d(+d|S)?a- 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) rNc2|dntj|S)N)npasarray)xs ]/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/graphics/dotplots.pyzdot_plot..s 1 g?FTrg{Gz?)rrg@gc<i|]}||S)index).0r style_codess r zdot_plot..s)DDD1q+++A..DDDr )rgbykpurpleorangeci|]}|iSrrrr s r rzdot_plot..s333!2333r colormarkeroms ci|]}|iSrrrs r rzdot_plot..s111a111r grey linewidthgffffff?darkgrey)r transformzordergffffff?center)horizontalalignmentverticalalignmentr*boldc,g|]}|dk|S)rr)rrk0s r zdot_plot..=s"<<> ^ ,r* * *(.q3v;;(?L W % <+ + +),L X & |B' ' '+1Q;;RRAL T "11[111 4[[AA ^ *R. ( (&,JrN7 # jn , ,/5zz!!qJrN; '#qjjcDFllctAvoeeG Ekk # +! +%'6!9__S46\\#B!$S)9%9!9!$S)9%9!9!;&0*,,'(  ***ggc3.>)>#>2:08(* 66""6***''<<<v#EEBBZ({9R=112#% " #52.! $& " a(8#82q!11!":c SVSV,rs^;?8<>B7<)/04*. ccccccr