love_plot

love_plot(x, designs, *, imbalance='tasmd', treat_lab=1, mean_tar=None, sd_tar=None, denom='target', vline=None, xupper=1.0, ax=None)[source]

Plot imbalance per covariate for one or two assignments.

Parameters:

  • x (pd.DataFrame) – Covariates.

  • designs (dict[str, np.ndarray]) – Named assignment vectors, one marker series each.

  • imbalance (Literal['tasmd', 'asmd']) – TASMD of group treat_lab, or ASMD between the two smallest labels.

  • treat_lab (object) – Group assessed by TASMD.

  • mean_tar (np.ndarray | None) – Target means for TASMD.

  • sd_tar (np.ndarray | None) – Target standard deviations for TASMD.

  • denom (Literal['target', 'group']) – Denominator for TASMD; see tasmd().

  • vline (float | None) – Draw a reference line at this imbalance.

  • xupper (float) – Upper limit of the imbalance axis.

  • ax (Axes | None) – Axes to draw on; a new figure is made if omitted.

Returns:

The axes drawn on.

Return type:

Axes