`bucky.viz.plot`#

Creates plots for Bucky data.

Module Contents#

Functions#

`_pool_plot`(args)	Function to create plots given a tuple of input data.
`add_simulation_data_to_axis`(axis, col, sim_data, quantiles)	Adds simulation data for requested column to initialized matplotlib axis object.
`default_plot`(cfg)	Creates default Bucky plot, one plot per unique ADM region containing as many subplots as columns requested.
`main`(cfg)	Main entrypoint.
`scale_y_axis`(axis, col, sim_data, quantiles, hist_data, fit_data, zero_as_ymin=True)	Adds simulation data for requested column to initialized matplotlib axis object.

bucky.viz.plot._pool_plot(args)[source]#

Function to create plots given a tuple of input data.

This input data should include a pandas DataFrameGroupBy object, the plot name, historical data in the same order as the keys in the GroupBy object, the columns and quantiles to plot, the admin key, and the directory to save plots and CSVs.

Parameters:: args (tuple) – Zipped input data

bucky.viz.plot.add_simulation_data_to_axis(axis, col, sim_data, quantiles)[source]#

Adds simulation data for requested column to initialized matplotlib axis object.

Parameters:

axis (matplotlib.axes.Axes) – Previously initialized axis object
col (str) – Column to add to plot
sim_data (pandas.DataFrame) – Area simulation data
quantiles (list of float, or None) – List of quantiles to plot. If None, will plot all available quantiles in data.

Returns:

axis – Modified axis object with added data

Return type:

matplotlib.axes.Axes

bucky.viz.plot.default_plot(cfg)[source]#

Creates default Bucky plot, one plot per unique ADM region containing as many subplots as columns requested.

Parameters:: cfg (BuckyConfig) – BuckyConfig object with various plot-related parameters

bucky.viz.plot.main(cfg)[source]#: Main entrypoint.

bucky.viz.plot.scale_y_axis(axis, col, sim_data, quantiles, hist_data, fit_data, zero_as_ymin=True)[source]#