AeroViz 0.1.3b0__py3-none-any.whl → 0.1.4__py3-none-any.whl

AeroViz/plot/timeseries/timeseries.py

@@ -1,38 +1,39 @@
-from datetime import datetime
 from typing import Literal
 
 import matplotlib.pyplot as plt
+import numpy as np
 from matplotlib.cm import ScalarMappable
 from matplotlib.pyplot import Figure, Axes
 from mpl_toolkits.axes_grid1.inset_locator import inset_axes
-from pandas import DataFrame, date_range, Timestamp
+from pandas import DataFrame, date_range
 
 from AeroViz.plot.utils import *
 
-__all__ = ['timeseries', 'timeseries_template']
+__all__ = ['timeseries', 'timeseries_stacked']
+
 
 default_bar_kws = dict(
-	width=0.0417,
-	edgecolor=None,
-	linewidth=0,
-	cmap='jet',
+    width=0.0417,
+    edgecolor=None,
+    linewidth=0,
+    cmap='jet',
 )
 
 default_scatter_kws = dict(
-	marker='o',
-	s=5,
-	edgecolor=None,
-	linewidths=0.3,
-	alpha=0.9,
-	cmap='jet',
+    marker='o',
+    s=5,
+    edgecolor=None,
+    linewidths=0.3,
+    alpha=0.9,
+    cmap='jet',
 )
 
 default_insert_kws = dict(
-	width="1.5%",
-	height="100%",
-	loc='lower left',
-	bbox_to_anchor=(1.01, 0, 1.2, 1),
-	borderpad=0
+    width="1.5%",
+    height="100%",
+    loc='lower left',
+    bbox_to_anchor=(1.01, 0, 1.2, 1),
+    borderpad=0
 )
 
 default_plot_kws = dict()
@@ -41,51 +42,62 @@ default_cbar_kws = dict()
 
 
 def _scatter(ax, df, _y, _c, scatter_kws, cbar_kws, inset_kws):
-	if _c is None or _c not in df.columns:
-		scatter_kws.pop('cmap')
-		ax.scatter(df.index, df[_y], **scatter_kws)
-	else:
-		ax.scatter(df.index, df[_y], c=df[_c], **scatter_kws)
-		cax = inset_axes(ax, **inset_kws)
+    if _c is None or _c not in df.columns:
+        scatter_kws.pop('cmap')
+        ax.scatter(df.index, df[_y], **scatter_kws)
+    else:
+        ax.scatter(df.index, df[_y], c=df[_c], **scatter_kws)
+        cax = inset_axes(ax, **inset_kws)
 
-		# Filter the children to find ScalarMappable objects
-		mappable_objects = [child for child in ax.get_children() if isinstance(child, ScalarMappable)]
+        # Filter the children to find ScalarMappable objects
+        mappable_objects = [child for child in ax.get_children() if isinstance(child, ScalarMappable)]
 
-		# Use the first mappable object for the colorbar
-		if mappable_objects:
-			plt.colorbar(mappable=mappable_objects[0], cax=cax, **cbar_kws)
-		else:
-			print("No mappable objects found.")
+        # Use the first mappable object for the colorbar
+        if mappable_objects:
+            plt.colorbar(mappable=mappable_objects[0], cax=cax, **cbar_kws)
+        else:
+            print("No mappable objects found.")
 
 
 def _bar(ax, df, _y, _c, bar_kws, cbar_kws, inset_kws):
-	scalar_map, colors = Color.color_maker(df[_c].values, cmap=bar_kws.pop('cmap'))
-	ax.bar(df.index, df[_y], color=scalar_map.to_rgba(colors), **bar_kws)
-	cax = inset_axes(ax, **inset_kws)
-	plt.colorbar(mappable=scalar_map, cax=cax, **cbar_kws)
+    scalar_map, colors = Color.color_maker(df[_c].values, cmap=bar_kws.pop('cmap'))
+    ax.bar(df.index, df[_y], color=scalar_map.to_rgba(colors), **bar_kws)
+    cax = inset_axes(ax, **inset_kws)
+    plt.colorbar(mappable=scalar_map, cax=cax, **cbar_kws)
 
 
 def _plot(ax, df, _y, _color, plot_kws):
-	ax.plot(df.index, df[_y], color=_color, **plot_kws)
+    ax.plot(df.index, df[_y], color=_color, **plot_kws)
+
+
+def process_timeseries_data(df, rolling=None, interpolate_limit=None):
+    # apply rolling window if specified
+    df = df.rolling(window=rolling, min_periods=1).mean(numeric_only=True) if rolling is not None else df
 
+    # apply interpolation if specified
+    df = df.interpolate(method='time', limit=interpolate_limit) if interpolate_limit is not None else df
+    return df
 
-@set_figure(fs=8, autolayout=False)
+
+@set_figure(autolayout=False)
 def timeseries(df: DataFrame,
-			   y: list[str] | str,
-			   y2: list[str] | str = None,
-			   c: list[str] | str = None,
-			   # color: list[str] | str = None,
-			   rolling: str | int | None = None,
-			   times: list[datetime | Timestamp | str] = None,
-			   freq: str = '1MS',
-			   style: list[Literal['scatter', 'bar', 'line']] | str | None = None,
-			   ax: Axes | None = None,
-			   set_xaxis_visible: bool | None = None,
-			   legend_loc: Literal['best', 'upper right', 'upper left', 'lower left', 'lower right'] = 'best',
-			   legend_ncol: int = 1,
-			   **kwargs
-			   ) -> tuple[Figure, Axes]:
-	"""
+               y: list[str] | str,
+               y2: list[str] | str = None,
+               yi: list[str] | str = None,
+               color: list[str] | str | None = None,
+               label: list[str] | str | None = None,
+               rolling: int | str | None = 3,
+               interpolate_limit: int | None = 6,
+               major_freq: str = '1MS',
+               minor_freq: str = '10d',
+               style: list[Literal['scatter', 'bar', 'line']] | str | None = None,
+               ax: Axes | None = None,
+               set_xaxis_visible: bool | None = None,
+               legend_loc: Literal['best', 'upper right', 'upper left', 'lower left', 'lower right'] = 'best',
+               legend_ncol: int = 1,
+               **kwargs
+               ) -> tuple[Figure, Axes]:
+    """
     Plot the timeseries data with the option of scatterplot, barplot, and lineplot.
 
     Parameters
@@ -96,14 +108,20 @@ def timeseries(df: DataFrame,
         The primary y-axis data columns.
     y2 : list[str] | str, optional
         The secondary y-axis data columns. Defaults to None.
-    c : str, optional
+    yi : list[str] | str, optional
+        The components for percentage calculation. Defaults to None.
+    color : str, optional
         The column for color mapping or the color. Defaults to None.
+    label : str, optional
+        The label for the legend. Defaults to None.
     rolling : str | int | None, optional
         Rolling window size for smoothing. Defaults to None.
-    times : tuple[datetime, datetime] | tuple[Timestamp, Timestamp], optional
-        Time range for the data. Defaults to None.
-    freq : str, optional
-        Frequency for x-axis ticks. Defaults to '2MS'.
+    interpolate_limit : int, optional
+        Interpolation limit for missing values. Defaults to None.
+    major_freq : str, optional
+        Frequency for x-axis ticks. Defaults to '1MS'.
+    minor_freq : str, optional
+        Frequency for x-axis minor ticks. Defaults to '10d'.
     style : Literal['scatter', 'bar', 'line'] | None, optional
         Style of the plot. Defaults to 'scatter'.
     ax : Axes | None, optional
@@ -137,159 +155,208 @@ def timeseries(df: DataFrame,
 
     Example
     -------
-    >>> timeseries(df, y='WS', c='WD', scatter_kws=dict(cmap='hsv'), cbar_kws=dict(ticks=[0, 90, 180, 270, 360]), ylim=[0, None])
+    >>> timeseries(df, y='WS', color='WD', scatter_kws=dict(cmap='hsv'), cbar_kws=dict(ticks=[0, 90, 180, 270, 360]), ylim=[0, None])
     """
-	# Set the time
+    # Set the time
+    try:
+        st_tm, fn_tm = df.index[0], df.index[-1]
+    except IndexError:
+        raise IndexError("The DataFrame is empty. Please provide a valid DataFrame.")
 
-	if times is not None:
-		st_tm, fn_tm = map(Timestamp, times)
-	else:
-		try:
-			st_tm, fn_tm = df.index[0], df.index[-1]
-		except IndexError:
-			raise IndexError("The DataFrame is empty. Please provide a valid DataFrame.")
+    # calculate the percentage of each component
+    if yi is not None:
+        df_pct = df[yi].div(df[yi].sum(axis=1), axis=0) * 100
+        mean = [f"{_label} : {df[comp].mean():.2f}" for _label, comp in zip(label, yi)]
+        pct = [f"{_label} : {df_pct[comp].mean():.2f}%" for _label, comp in zip(label, yi)]
+        df_pct = process_timeseries_data(df_pct, rolling, interpolate_limit)
 
-	# Apply rolling window if specified
-	df = df.loc[st_tm:fn_tm] if rolling is None else (
-		df.loc[st_tm:fn_tm].rolling(window=rolling, min_periods=1).mean(numeric_only=True))
+    # process data
+    df = process_timeseries_data(df, rolling, interpolate_limit)
 
-	# Initialize figure and axis if not provided
-	fig, ax = plt.subplots(**{**{'figsize': (6, 2)}, **kwargs.get('fig_kws', {})}) if ax is None else (
-		ax.get_figure(), ax)
+    # Initialize figure and axis if not provided
+    fig, ax = plt.subplots(**{**{'figsize': (6, 2)}, **kwargs.get('fig_kws', {})}) if ax is None else (
+        ax.get_figure(), ax)
 
-	# Ensure y, y2, c, and style are lists
-	y = [y] if isinstance(y, str) else y
-	y2 = [y2] if isinstance(y2, str) else y2 if y2 is not None else []
-	c = [c] if isinstance(c, str) else c if c is not None else [None] * (len(y) + len(y2))
-	style = [style] if isinstance(style, str) else style if style is not None else ['plot'] * (len(y) + len(y2))
-
-	if len(c) != len(y) + len(y2):
-		raise ValueError("The length of c must match the combined length of y and y2")
-
-	if len(style) != len(y) + len(y2):
-		raise ValueError("The length of style must match the combined length of y and y2")
-
-	# Create a secondary y-axis if y2 is not empty
-	ax2 = ax.twinx() if y2 else None
-
-	# # Set color cycle
-	ax.set_prop_cycle(Color.color_cycle)
-	if y2:
-		ax2.set_prop_cycle(Color.color_cycle[len(y):])
-
-	if y2 and ('scatter' or 'bar') in style:
-		fig.subplots_adjust(right=0.8)
-
-	for i, _c in enumerate(c):
-		if _c is not None and _c in df.columns:
-			style[i] = 'scatter'
-
-	for i, (_y, _c, _style) in enumerate(zip(y, c, style)):
-		scatter_kws = {**default_scatter_kws, **{'label': Unit(_y)}, **kwargs.get('scatter_kws', {})}
-		bar_kws = {**default_bar_kws, **{'label': Unit(_y)}, **kwargs.get('bar_kws', {})}
-		plot_kws = {**default_plot_kws, **{'label': Unit(_y)}, **kwargs.get('plot_kws', {})}
-
-		if _style in ['scatter', 'bar']:
-			cbar_kws = {**default_cbar_kws, **{'label': Unit(_c), 'ticks': None}, **kwargs.get('cbar_kws', {})}
-			inset_kws = {**default_insert_kws, **{'bbox_transform': ax.transAxes}, **kwargs.get('inset_kws', {})}
-
-		if _style == 'scatter':
-			_scatter(ax, df, _y, _c, scatter_kws, cbar_kws, inset_kws)
-
-		elif _style == 'bar':
-			_bar(ax, df, _y, _c, bar_kws, cbar_kws, inset_kws)
-
-		else:
-			_plot(ax, df, _y, _c, plot_kws)
-
-	if y2:
-		for i, (_y, _c, _style) in enumerate(zip(y2, c[len(y):], style[len(y):])):
-			scatter_kws = {**default_scatter_kws, **{'label': Unit(_y)}, **kwargs.get('scatter_kws2', {})}
-			bar_kws = {**default_bar_kws, **{'label': Unit(_y)}, **kwargs.get('bar_kws2', {})}
-			plot_kws = {**default_plot_kws, **{'label': Unit(_y)}, **kwargs.get('plot_kws2', {})}
-
-			if _style in ['scatter', 'bar']:
-				cbar_kws = {**default_cbar_kws, **{'label': Unit(_c), 'ticks': None}, **kwargs.get('cbar_kws2', {})}
-				inset_kws = {**default_insert_kws, **{'bbox_transform': ax.transAxes}, **kwargs.get('inset_kws2', {})}
-
-			if _style == 'scatter':
-				_scatter(ax2, df, _y, _c, scatter_kws, cbar_kws, inset_kws)
-
-			elif _style == 'bar':
-				_bar(ax2, df, _y, _c, bar_kws, cbar_kws, inset_kws)
-
-			else:  # line plot
-				_plot(ax2, df, _y, _c, plot_kws)
-
-		# Combine legends from ax and ax2
-		ax.legend(*combine_legends([ax, ax2]), loc=legend_loc, ncol=legend_ncol)
-
-	else:
-		ax.legend(loc=legend_loc, ncol=legend_ncol)
-
-	if set_xaxis_visible is not None:
-		ax.axes.xaxis.set_visible(set_xaxis_visible)
-
-	ax.set(xlabel=kwargs.get('xlabel', ''),
-		   ylabel=kwargs.get('ylabel', Unit(y) if isinstance(y, str) else Unit(y[0])),
-		   xticks=kwargs.get('xticks', date_range(start=st_tm, end=fn_tm, freq=freq).strftime("%F")),
-		   yticks=kwargs.get('yticks', ax.get_yticks()),
-		   xticklabels=kwargs.get('xticklabels', date_range(start=st_tm, end=fn_tm, freq=freq).strftime("%F")),
-		   yticklabels=kwargs.get('yticklabels', ax.get_yticklabels()),
-		   xlim=kwargs.get('xlim', (st_tm, fn_tm)),
-		   ylim=kwargs.get('ylim', (None, None)),
-		   title=kwargs.get('title', '')
-		   )
-
-	if y2:
-		ax2.set(ylabel=kwargs.get('ylabel2', Unit(y2) if isinstance(y2, str) else Unit(y2[0])),
-				yticks=kwargs.get('yticks2', ax2.get_yticks()),
-				yticklabels=kwargs.get('yticklabels2', ax2.get_yticklabels()),
-				ylim=kwargs.get('ylim2', (None, None)))
-
-	plt.show()
-
-	return fig, ax
-
-
-@set_figure(fs=8, autolayout=False)
-def timeseries_template(df: DataFrame) -> tuple[Figure, Axes]:
-	fig, ax = plt.subplots(5, 1, figsize=(len(df.index) * 0.01, 4))
-	(ax1, ax2, ax3, ax4, ax5) = ax
-
-	timeseries(df,
-			   y=['Extinction', 'Scattering', 'Absorption'],
-			   rolling=30,
-			   ax=ax1,
-			   ylabel='Coefficient',
-			   ylim=[0., None],
-			   set_xaxis_visible=False,
-			   legend_ncol=3,
-			   )
-
-	# Temp, RH
-	timeseries(df,
-			   y='AT',
-			   y2='RH',
-			   rolling=30,
-			   ax=ax2,
-			   ax_plot_kws=dict(color='r'),
-			   ax2_plot_kws=dict(color='b'),
-			   ylim=[10, 30],
-			   ylim2=[20, 100],
-			   set_xaxis_visible=False,
-			   legend_ncol=2,
-			   )
-
-	timeseries(df, y='WS', c='WD', style='scatter', ax=ax3, scatter_kws=dict(cmap='hsv'),
-			   cbar_kws=dict(ticks=[0, 90, 180, 270, 360]),
-			   ylim=[0, None], set_xaxis_visible=False)
-
-	timeseries(df, y='VC', c='PBLH', style='bar', ax=ax4, bar_kws=dict(cmap='Blues'), set_xaxis_visible=False,
-			   ylim=[0, 5000])
-
-	timeseries(df, y='PM25', c='PM1/PM25', style='scatter', ax=ax5, ylim=[0, None])
-
-	plt.show()
-
-	return fig, ax
+    # Ensure y, y2, c, and style are lists
+    y = [y] if isinstance(y, str) else y
+    y2 = [y2] if isinstance(y2, str) else y2 if y2 is not None else []
+    color = [color] if isinstance(color, str) else color if color is not None else [None] * (len(y) + len(y2))
+    label = [label] if isinstance(label, str) else label if label is not None else [None] * (len(y) + len(y2))
+    style = [style] if isinstance(style, str) else style if style is not None else ['plot'] * (len(y) + len(y2))
+
+    for name, lst in [("c", color), ("style", style), ("label", label)]:
+        if len(lst) != len(y) + len(y2):
+            raise ValueError(f"The length of {name} must match the combined length of y and y2")
+
+    # Create a secondary y-axis if y2 is not empty
+    ax2 = ax.twinx() if y2 else None
+
+    # # Set color cycle
+    ax.set_prop_cycle(Color.color_cycle)
+    if y2:
+        ax2.set_prop_cycle(Color.color_cycle[len(y):])
+
+    if y2 and ('scatter' or 'bar') in style:
+        fig.subplots_adjust(right=0.8)
+
+    for i, _c in enumerate(color):
+        if _c is not None and _c in df.columns:
+            style[i] = 'scatter'
+
+    for i, (_y, _c, _label, _style) in enumerate(zip(y, color, label, style)):
+        scatter_kws = {**default_scatter_kws, **{'label': Unit(_y)}, **kwargs.get('scatter_kws', {})}
+        bar_kws = {**default_bar_kws, **{'label': Unit(_y)}, **kwargs.get('bar_kws', {})}
+        plot_kws = {**default_plot_kws, **{'label': Unit(_y)}, **kwargs.get('plot_kws', {})}
+
+        if _style in ['scatter', 'bar']:
+            cbar_kws = {**default_cbar_kws, **{'label': Unit(_c), 'ticks': None}, **kwargs.get('cbar_kws', {})}
+            inset_kws = {**default_insert_kws, **{'bbox_transform': ax.transAxes}, **kwargs.get('inset_kws', {})}
+
+        if _style == 'scatter':
+            _scatter(ax, df, _y, _c, scatter_kws, cbar_kws, inset_kws)
+
+        elif _style == 'bar':
+            _bar(ax, df, _y, _c, bar_kws, cbar_kws, inset_kws)
+
+        else:
+            _plot(ax, df, _y, _c, plot_kws)
+
+    if y2:
+        for i, (_y, _c, _style) in enumerate(zip(y2, color[len(y):], style[len(y):])):
+            scatter_kws = {**default_scatter_kws, **{'label': Unit(_y)}, **kwargs.get('scatter_kws2', {})}
+            bar_kws = {**default_bar_kws, **{'label': Unit(_y)}, **kwargs.get('bar_kws2', {})}
+            plot_kws = {**default_plot_kws, **{'label': Unit(_y)}, **kwargs.get('plot_kws2', {})}
+
+            if _style in ['scatter', 'bar']:
+                cbar_kws = {**default_cbar_kws, **{'label': Unit(_c), 'ticks': None}, **kwargs.get('cbar_kws2', {})}
+                inset_kws = {**default_insert_kws, **{'bbox_transform': ax.transAxes}, **kwargs.get('inset_kws2', {})}
+
+            if _style == 'scatter':
+                _scatter(ax2, df, _y, _c, scatter_kws, cbar_kws, inset_kws)
+
+            elif _style == 'bar':
+                _bar(ax2, df, _y, _c, bar_kws, cbar_kws, inset_kws)
+
+            else:  # line plot
+                _plot(ax2, df, _y, _c, plot_kws)
+
+        # Combine legends from ax and ax2
+        ax.legend(*combine_legends([ax, ax2]), loc=legend_loc, ncol=legend_ncol)
+
+    else:
+        ax.legend(loc=legend_loc, ncol=legend_ncol)
+
+    if set_xaxis_visible is not None:
+        ax.axes.xaxis.set_visible(set_xaxis_visible)
+
+    ax.set(xlabel=kwargs.get('xlabel', ''),
+           ylabel=kwargs.get('ylabel', Unit(y) if isinstance(y, str) else Unit(y[0])),
+           xlim=kwargs.get('xlim', (st_tm, fn_tm)),
+           ylim=kwargs.get('ylim', (None, None)),
+           title=kwargs.get('title', '')
+           )
+
+    xticks = kwargs.get('xticks', date_range(start=st_tm, end=fn_tm, freq=major_freq))
+    minor_xticks = kwargs.get('minor_xticks', date_range(start=st_tm, end=fn_tm, freq=minor_freq))
+
+    ax.set_xticks(ticks=xticks, labels=xticks.strftime("%F"))
+    ax.set_xticks(minor_xticks, minor=True)
+
+    if y2:
+        ax2.set(ylim=kwargs.get('ylim2', (None, None)),
+                ylabel=kwargs.get('ylabel2', Unit(y2) if isinstance(y2, str) else Unit(y2[0]))
+                )
+
+    plt.show()
+
+    return fig, ax
+
+
+@set_figure(autolayout=False)
+def timeseries_stacked(df,
+                       y: list[str] | str,
+                       yi: list[str] | str,
+                       label: list[str] | str,
+                       rolling: int | str | None = 6,
+                       interpolate_limit: int | None = 6,
+                       major_freq: str = '1MS',
+                       minor_freq: str = '10d',
+                       ax: Axes | None = None,
+                       legend_ncol: int = 1,
+                       **kwargs
+                       ) -> tuple[Figure, Axes]:
+    try:
+        st_tm, fn_tm = df.index[0], df.index[-1]
+    except IndexError:
+        raise IndexError("The DataFrame is empty. Please provide a valid DataFrame.")
+
+    # calculate the percentage of each component
+    df_pct = df[yi].div(df[yi].sum(axis=1), axis=0) * 100
+    mean = [f"{_label} : {df[comp].mean():.2f}" for _label, comp in zip(label, yi)]
+    pct = [f"{_label} : {df_pct[comp].mean():.2f}%" for _label, comp in zip(label, yi)]
+
+    # process data
+    df = process_timeseries_data(df, rolling, interpolate_limit)
+    df_pct = process_timeseries_data(df_pct, rolling, interpolate_limit)
+
+    fig, (ax1, ax2) = plt.subplots(2, 1, **{**{'figsize': (12, 3)}, **kwargs.get('fig_kws', {})})
+
+    width = 0.0417
+    color = Color.colors1
+
+    for name, lst in [("color", color), ("label", label)]:
+        if len(lst) != len(yi):
+            raise ValueError(f"The length of {name} must match the combined length of y and y2")
+
+    bottom = None  # 初始化堆疊位置
+    for i, (_column, _color, _label) in enumerate(zip(yi, color, mean)):
+        if i == 0:
+            bottom = df[_column] * 0  # 第一個堆疊底部為零
+        ax1.bar(df.index, df[_column], color=_color, width=width, bottom=bottom, label=_label)
+        bottom += df[_column]  # 更新堆疊底部位置
+
+    ax1.legend(loc='upper left', ncol=legend_ncol, prop={'weight': 'bold'}, bbox_to_anchor=(1, 0, 1.2, 1))
+
+    ax1.axes.xaxis.set_visible(False)
+
+    ax1.set(xlabel=kwargs.get('xlabel', ''),
+            ylabel=kwargs.get('ylabel', Unit(y) if isinstance(y, str) else Unit(y[0])),
+            xlim=kwargs.get('xlim', (st_tm, fn_tm)),
+            ylim=kwargs.get('ylim', (None, None)),
+            title=kwargs.get('title', ''),
+            )
+
+    xticks = kwargs.get('xticks', date_range(start=st_tm, end=fn_tm, freq=major_freq))
+    yticks = kwargs.get('yticks', np.linspace(*ax1.get_ylim(), num=6))
+    minor_xticks = kwargs.get('minor_xticks', date_range(start=st_tm, end=fn_tm, freq=minor_freq))
+
+    ax1.set_xticks(ticks=xticks, labels=xticks.strftime("%F"))
+    ax1.set_yticks(ticks=yticks, labels=[f'{tick:.0f}' for tick in yticks])
+    ax1.set_xticks(minor_xticks, minor=True)
+
+    # ax2
+    bottom = None  # 初始化堆疊位置
+    for i, (_column, _color, _label) in enumerate(zip(yi, color, pct)):
+        if i == 0:
+            bottom = df_pct[_column] * 0  # 第一個堆疊底部為零
+        ax2.bar(df_pct.index, df_pct[_column], color=_color, width=width, bottom=bottom, label=_label)
+        bottom += df_pct[_column]  # 更新堆疊底部位置
+
+    ax2.legend(loc='upper left', ncol=legend_ncol, prop={'weight': 'bold'}, bbox_to_anchor=(1, 0, 1.2, 1))
+
+    ax2.set(xlabel=kwargs.get('xlabel', ''),
+            ylabel=kwargs.get('ylabel', 'Percentage (%)'),
+            xlim=kwargs.get('xlim', (st_tm, fn_tm)),
+            ylim=(0, 100),
+            title=kwargs.get('title', '')
+            )
+
+    xticks = kwargs.get('xticks', date_range(start=st_tm, end=fn_tm, freq=major_freq))
+    yticks = kwargs.get('yticks', np.linspace(*ax2.get_ylim(), num=6))
+    minor_xticks = kwargs.get('minor_xticks', date_range(start=st_tm, end=fn_tm, freq=minor_freq))
+
+    ax2.set_xticks(ticks=xticks, labels=xticks.strftime("%F"))
+    ax2.set_yticks(ticks=yticks, labels=[f'{tick:.0f}' for tick in yticks])
+    ax2.set_xticks(minor_xticks, minor=True)
+
+    return fig, ax1

AeroViz/plot/utils/plt_utils.py

@@ -17,7 +17,7 @@ def set_figure(func=None,
     def decorator(_func):
         @wraps(_func)
         def wrapper(*args, **kwargs):
-            print(f'\tPlot:\033[96m {_func.__name__}\033[0m')
+            print(f'\n\tPlot:\033[96m {_func.__name__}\033[0m')
 
             plt.rcParams['mathtext.fontset'] = 'custom'
             plt.rcParams['mathtext.rm'] = 'Times New Roman'
@@ -53,7 +53,7 @@ def set_figure(func=None,
             plt.rcParams['figure.dpi'] = 200
             plt.rcParams['figure.autolayout'] = autolayout
 
-            if ~autolayout:
+            if not autolayout:
                 plt.rcParams['figure.subplot.left'] = 0.1
                 plt.rcParams['figure.subplot.right'] = 0.875
                 plt.rcParams['figure.subplot.top'] = 0.875

AeroViz/plot/utils/units.json

@@ -6,6 +6,8 @@
   "T_OC": "OC (\u00b5g/m^3)",
   "PM1": "PM_{1} (\u00b5g/m^3)",
   "PM25": "PM_{2.5} (\u00b5g/m^3)",
+  "PM2.5": "PM_{2.5} (\u00b5g/m^3)",
+  "PM10": "PM_{10} (\u00b5g/m^3)",
   "SIA": "SIA (\u00b5g/m^3)",
   "POC": "POC (\u00b5g/m^3)",
   "SOC": "SOC (\u00b5g/m^3)",
@@ -20,12 +22,15 @@
   "Babs": "Mie Amb Absorption (1/Mm)",
   "Babs_dry": "Mie Dry Absorption (1/Mm)",
   "Absorption": "Absorption (1/Mm)",
+  "abs": "Absorption (1/Mm)",
   "Bext": "Mie Amb Extinction (1/Mm)",
   "Bext_dry": "Mie Dry Extinction (1/Mm)",
   "Extinction": "Extinction (1/Mm)",
+  "ext": "Extinction (1/Mm)",
   "Bsca": "Mie Amb Scattering (1/Mm)",
   "Bsca_dry": "Mie Dry Scattering (1/Mm)",
   "Scattering": "Scattering (1/Mm)",
+  "sca": "Scattering (1/Mm)",
   "Diurnal": "Hour",
   "PBLH": "PBLH (m)",
   "VC": "VC (m²/s)",

AeroViz/plot/violin.py

@@ -1,6 +1,7 @@
 import matplotlib.pyplot as plt
 import numpy as np
 import pandas as pd
+import seaborn as sns
 from matplotlib.pyplot import Figure, Axes
 from pandas import DataFrame
 
@@ -10,7 +11,7 @@ __all__ = ['violin']
 
 
 @set_figure(fw='bold')
-def violin(data_set: DataFrame | dict,
+def violin(df: DataFrame | dict,
            unit: str,
            ax: Axes | None = None,
            **kwargs
@@ -20,7 +21,7 @@ def violin(data_set: DataFrame | dict,
 
     Parameters
     ----------
-    data_set : pd.DataFrame or dict
+    df : pd.DataFrame or dict
         A mapping from category names to pandas DataFrames containing the data.
     unit : str
         The unit for the data being plotted.
@@ -31,13 +32,15 @@ def violin(data_set: DataFrame | dict,
 
     Returns
     -------
-    matplotlib.axes.Axes
-        The Axes object containing the violin plot.
+    fig : Figure
+        The matplotlib Figure object.
+    ax : Axes
+        The matplotlib Axes object with the scatter plot.
 
     """
     fig, ax = plt.subplots(**kwargs.get('fig_kws', {})) if ax is None else (ax.get_figure(), ax)
 
-    data = data_set.to_numpy()
+    data = df.to_numpy()
 
     data = data[~np.isnan(data).any(axis=1)]
 
@@ -67,13 +70,11 @@ def violin(data_set: DataFrame | dict,
     ylim = kwargs.get('ylim') or (0, None)
     xlabel = kwargs.get('xlabel') or ''
     ylabel = kwargs.get('ylabel') or Unit(unit)
-    xticks = kwargs.get('xticks') or [x.replace('-', '\n') for x in list(data_set.keys())]
+    xticks = kwargs.get('xticks') or [x.replace('-', '\n') for x in list(df.keys())]
 
     ax.set(xlim=xlim, ylim=ylim, xlabel=xlabel, ylabel=ylabel, title=kwargs.get('title'))
     ax.set_xticks(x_position, xticks, fontweight='bold', fontsize=12)
 
-    # fig.savefig(f'Violin_{unit}')
-
     plt.show()
 
     return fig, ax