idmd.visualization.plots.PlotGenerator

class idmd.visualization.plots.PlotGenerator[source]

Bases: object

Generates various types of plots.

Methods

generate_bar_plot

Generates a bar plot for the specified column.

generate_line_plot

Generates a line plot for the specified columns.
static generate_bar_plot(df, column)[source]

Generates a bar plot for the specified column.

Return type:

Figure

Args:

df (pd.DataFrame): The DataFrame containing the data. column (str): The column to include in the bar plot.

Returns:

plt.Figure: The generated bar plot.

static generate_line_plot(df, columns)[source]

Generates a line plot for the specified columns.

Return type:

Figure

Args:

df (pd.DataFrame): The DataFrame containing the data. columns (List[str]): The columns to include in the line plot.

Returns:

plt.Figure: The generated line plot.