44 < head >
55 < meta charset ="utf-8 " />
66 < meta name ="viewport " content ="width=device-width, initial-scale=1.0 " />
7- < title > dte_adj.plot — dte_adj 0.1.7 documentation</ title >
7+ < title > dte_adj.plot — dte_adj 0.1.8 documentation</ title >
88 < link rel ="stylesheet " type ="text/css " href ="../../_static/pygments.css?v=5ecbeea2 " />
99 < link rel ="stylesheet " type ="text/css " href ="../../_static/alabaster.css?v=12dfc556 " />
10- < script src ="../../_static/documentation_options.js?v=ca7ad2ea "> </ script >
10+ < script src ="../../_static/documentation_options.js?v=22607128 "> </ script >
1111 < script src ="../../_static/doctools.js?v=9a2dae69 "> </ script >
1212 < script src ="../../_static/sphinx_highlight.js?v=dc90522c "> </ script >
1313 < link rel ="index " title ="Index " href ="../../genindex.html " />
@@ -49,6 +49,7 @@ <h1>Source code for dte_adj.plot</h1><div class="highlight"><pre>
4949 < span class ="n "> title</ span > < span class ="p "> :</ span > < span class ="n "> Optional</ span > < span class ="p "> [</ span > < span class ="nb "> str</ span > < span class ="p "> ]</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span > < span class ="p "> ,</ span >
5050 < span class ="n "> xlabel</ span > < span class ="p "> :</ span > < span class ="n "> Optional</ span > < span class ="p "> [</ span > < span class ="nb "> str</ span > < span class ="p "> ]</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span > < span class ="p "> ,</ span >
5151 < span class ="n "> ylabel</ span > < span class ="p "> :</ span > < span class ="n "> Optional</ span > < span class ="p "> [</ span > < span class ="nb "> str</ span > < span class ="p "> ]</ span > < span class ="o "> =</ span > < span class ="kc "> None</ span > < span class ="p "> ,</ span >
52+ < span class ="n "> weighted</ span > < span class ="p "> :</ span > < span class ="nb "> bool</ span > < span class ="o "> =</ span > < span class ="kc "> False</ span > < span class ="p "> ,</ span >
5253< span class ="p "> ):</ span >
5354< span class ="w "> </ span > < span class ="sd "> """Visualize distributional parameters and their confidence intervals.</ span >
5455
@@ -63,12 +64,18 @@ <h1>Source code for dte_adj.plot</h1><div class="highlight"><pre>
6364< span class ="sd "> title (str, optional): Axes title.</ span >
6465< span class ="sd "> xlabel (str, optional): X-axis title label.</ span >
6566< span class ="sd "> ylabel (str, optional): Y-axis title label.</ span >
67+ < span class ="sd "> weighted (bool, optional): If True, multiply treatment effects by X values to show value-weighted effects. Defaults to False.</ span >
6668
6769< span class ="sd "> Returns:</ span >
6870< span class ="sd "> matplotlib.axes.Axes: The axes with the plot.</ span >
6971< span class ="sd "> """</ span >
7072 < span class ="k "> if</ span > < span class ="n "> ax</ span > < span class ="ow "> is</ span > < span class ="kc "> None</ span > < span class ="p "> :</ span >
71- < span class ="n "> fig</ span > < span class ="p "> ,</ span > < span class ="n "> ax</ span > < span class ="o "> =</ span > < span class ="n "> plt</ span > < span class ="o "> .</ span > < span class ="n "> subplots</ span > < span class ="p "> ()</ span >
73+ < span class ="n "> _</ span > < span class ="p "> ,</ span > < span class ="n "> ax</ span > < span class ="o "> =</ span > < span class ="n "> plt</ span > < span class ="o "> .</ span > < span class ="n "> subplots</ span > < span class ="p "> ()</ span >
74+
75+ < span class ="k "> if</ span > < span class ="n "> weighted</ span > < span class ="p "> :</ span >
76+ < span class ="n "> means</ span > < span class ="o "> =</ span > < span class ="n "> means</ span > < span class ="o "> *</ span > < span class ="n "> X</ span >
77+ < span class ="n "> lower_bounds</ span > < span class ="o "> =</ span > < span class ="n "> lower_bounds</ span > < span class ="o "> *</ span > < span class ="n "> X</ span >
78+ < span class ="n "> upper_bounds</ span > < span class ="o "> =</ span > < span class ="n "> upper_bounds</ span > < span class ="o "> *</ span > < span class ="n "> X</ span >
7279
7380 < span class ="k "> if</ span > < span class ="n "> chart_type</ span > < span class ="o "> ==</ span > < span class ="s2 "> "line"</ span > < span class ="p "> :</ span >
7481 < span class ="n "> ax</ span > < span class ="o "> .</ span > < span class ="n "> plot</ span > < span class ="p "> (</ span > < span class ="n "> X</ span > < span class ="p "> ,</ span > < span class ="n "> means</ span > < span class ="p "> ,</ span > < span class ="n "> label</ span > < span class ="o "> =</ span > < span class ="s2 "> "Values"</ span > < span class ="p "> ,</ span > < span class ="n "> color</ span > < span class ="o "> =</ span > < span class ="n "> color</ span > < span class ="p "> )</ span >
0 commit comments