diff --git a/forestplot/mplot.py b/forestplot/mplot.py index 5598f5a..bcc29d0 100644 --- a/forestplot/mplot.py +++ b/forestplot/mplot.py @@ -78,6 +78,7 @@ def mforestplot( return_df: bool = False, preprocess: bool = True, table: bool = False, + legend: bool = True, **kwargs: Any, ) -> Axes: """ @@ -197,6 +198,7 @@ def mforestplot( yticker2=yticker2, color_alt_rows=color_alt_rows, table=table, + legend=legend, **kwargs, ) if return_df: @@ -309,6 +311,7 @@ def _make_mforestplot( despine: bool = True, color_alt_rows: bool = False, table: bool = False, + legend: bool = True, **kwargs: Any, ) -> Axes: if models is None: @@ -339,7 +342,10 @@ def _make_mforestplot( ax=ax, **kwargs, ) - ax = mdraw_legend(models=models, modellabels=modellabels, ax=ax, xlabel=xlabel, **kwargs) + if legend: + ax = mdraw_legend( + models=models, modellabels=modellabels, ax=ax, xlabel=xlabel, **kwargs + ) format_xticks( dataframe=dataframe, estimate=estimate, ll=ll, hl=hl, xticks=xticks, ax=ax, **kwargs diff --git a/forestplot/mplot_dataframe_utils.py b/forestplot/mplot_dataframe_utils.py index 079ba29..d387d1c 100644 --- a/forestplot/mplot_dataframe_utils.py +++ b/forestplot/mplot_dataframe_utils.py @@ -197,7 +197,7 @@ def make_multimodel_tableheaders( left_headers, right_headers, ) - dataframe.loc[ix, "model"] = models[c] + dataframe.loc[ix, model_col] = models[c] c += 1 return dataframe diff --git a/setup.py b/setup.py index 6ba8fed..6293fce 100644 --- a/setup.py +++ b/setup.py @@ -10,7 +10,7 @@ install_requires = ["pandas", "numpy", "matplotlib", "matplotlib-inline<=0.1.3"] setup( name="forestplot", - version="0.3.1", + version="0.4.0", license="MIT", author="Lucas Shen", author_email="lucas@lucasshen.com",