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

AeroViz/plot/timeseries/timeseries.py

@@ -69,13 +69,6 @@ def _plot(ax, df, _y, _color, plot_kws):
 	ax.plot(df.index, df[_y], color=_color, **plot_kws)
 
 
-def combine_legends(axes_list: list[Axes]) -> tuple[list, list]:
-	return (
-		[legend for axes in axes_list for legend in axes.get_legend_handles_labels()[0]],
-		[label for axes in axes_list for label in axes.get_legend_handles_labels()[1]]
-	)
-
-
 @set_figure(fs=8, autolayout=False)
 def timeseries(df: DataFrame,
 			   y: list[str] | str,
@@ -93,59 +86,59 @@ def timeseries(df: DataFrame,
 			   **kwargs
 			   ) -> tuple[Figure, Axes]:
 	"""
-	Plot the timeseries data with the option of scatterplot, barplot, and lineplot.
-
-	Parameters
-	-----------
-	df : DataFrame
-	The data to plot.
-	y : list[str] | str
-		The primary y-axis data columns.
-	y2 : list[str] | str, optional
-		The secondary y-axis data columns. Defaults to None.
-	c : str, optional
-		The column for color mapping or the color. 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'.
-	style : Literal['scatter', 'bar', 'line'] | None, optional
-		Style of the plot. Defaults to 'scatter'.
-	ax : Axes | None, optional
-		Matplotlib Axes object to plot on. Defaults to None.
-	set_xaxis_visible : bool | None, optional
-		Whether to set x-axis visibility. Defaults to None.
-	legend_loc : Literal['best', 'upper right', 'upper left', 'lower left', 'lower right'], optional
-		Location of the legend. Defaults to 'best'.
-	legend_ncol : int, optional
-		Number of columns in the legend. Defaults to 1.
-	**kwargs : Additional keyword arguments for customization.
-		fig_kws : dict, optional
-			Additional keyword arguments for the figure. Defaults to {}.
-		scatter_kws : dict, optional
-			Additional keyword arguments for the scatter plot. Defaults to {}.
-		bar_kws : dict, optional
-			Additional keyword arguments for the bar plot. Defaults to {}.
-		ax_plot_kws : dict, optional
-			Additional keyword arguments for the primary y-axis plot. Defaults to {}.
-		ax2_plot_kws : dict, optional
-			Additional keyword arguments for the secondary y-axis plot. Defaults to {}.
-		cbar_kws : dict, optional
-			Additional keyword arguments for the colorbar. Defaults to {}.
-		inset_kws : dict, optional
-			Additional keyword arguments for the inset axes. Defaults to {}.
-
-	Returns
-	-------
-	ax : AxesSubplot
-		Matplotlib AxesSubplot.
-
-	Example
-	-------
-	>>> timeseries(df, y='WS', c='WD', scatter_kws=dict(cmap='hsv'), cbar_kws=dict(ticks=[0, 90, 180, 270, 360]), ylim=[0, None])
-	"""
+    Plot the timeseries data with the option of scatterplot, barplot, and lineplot.
+
+    Parameters
+    -----------
+    df : DataFrame
+    The data to plot.
+    y : list[str] | str
+        The primary y-axis data columns.
+    y2 : list[str] | str, optional
+        The secondary y-axis data columns. Defaults to None.
+    c : str, optional
+        The column for color mapping or the color. 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'.
+    style : Literal['scatter', 'bar', 'line'] | None, optional
+        Style of the plot. Defaults to 'scatter'.
+    ax : Axes | None, optional
+        Matplotlib Axes object to plot on. Defaults to None.
+    set_xaxis_visible : bool | None, optional
+        Whether to set x-axis visibility. Defaults to None.
+    legend_loc : Literal['best', 'upper right', 'upper left', 'lower left', 'lower right'], optional
+        Location of the legend. Defaults to 'best'.
+    legend_ncol : int, optional
+        Number of columns in the legend. Defaults to 1.
+    **kwargs : Additional keyword arguments for customization.
+        fig_kws : dict, optional
+            Additional keyword arguments for the figure. Defaults to {}.
+        scatter_kws : dict, optional
+            Additional keyword arguments for the scatter plot. Defaults to {}.
+        bar_kws : dict, optional
+            Additional keyword arguments for the bar plot. Defaults to {}.
+        ax_plot_kws : dict, optional
+            Additional keyword arguments for the primary y-axis plot. Defaults to {}.
+        ax2_plot_kws : dict, optional
+            Additional keyword arguments for the secondary y-axis plot. Defaults to {}.
+        cbar_kws : dict, optional
+            Additional keyword arguments for the colorbar. Defaults to {}.
+        inset_kws : dict, optional
+            Additional keyword arguments for the inset axes. Defaults to {}.
+
+    Returns
+    -------
+    ax : AxesSubplot
+        Matplotlib AxesSubplot.
+
+    Example
+    -------
+    >>> timeseries(df, y='WS', c='WD', scatter_kws=dict(cmap='hsv'), cbar_kws=dict(ticks=[0, 90, 180, 270, 360]), ylim=[0, None])
+    """
 	# Set the time
 
 	if times is not None:

AeroViz/plot/utils/__init__.py

@@ -1,3 +1,4 @@
 from ._color import Color
-from ._decorator import set_figure
 from ._unit import Unit
+from .plt_utils import *
+from .sklearn_utils import *

AeroViz/plot/utils/_color.py

@@ -9,63 +9,63 @@ __all__ = ['Color']
 
 
 class Color:
-	color_cycle = cycler(color=['b', 'g', 'r', 'c', 'm', 'y', 'k'])
-
-	linecolor = [{'line': '#1a56db', 'edge': '#0F50A6', 'face': '#5983D9'},
-				 {'line': '#046c4e', 'edge': '#1B591F', 'face': '#538C4A'},
-				 {'line': '#c81e1e', 'edge': '#f05252', 'face': '#f98080'}]
-
-	# colors = ['#FF3333', '#33FF33', '#FFFF33', '#5555FF', '#B94FFF', '#AAAAAA', '#748690'] # the last one is "unknown"
-
-	colors1 = ['#A65E58', '#A5BF6B', '#F2BF5E', '#3F83BF', '#B777C2', '#D1CFCB']
-	colors2 = ['#A65E58', '#A5BF6B', '#F2BF5E', '#3F83BF', '#B777C2', '#D1CFCB', '#96c8e6']
-	colors3 = ['#A65E58', '#A5BF6B', '#a6710d', '#F2BF5E', '#3F83BF', '#B777C2', '#D1CFCB', '#96c8e6']  # POC SOC
-
-	colors_mutiWater = ['#A65E58', '#c18e8a', '#A5BF6B', '#c5d6a0', '#F2BF5E', '#3F83BF', '#c089ca', '#d3acda',
-						'#D1CFCB']
-	colors_mutiWater2 = ['#A65E58', '#96c8e6', '#A5BF6B', '#96c8e6', '#F2BF5E', '#3F83BF', '#c089ca', '#96c8e6',
-						 '#D1CFCB']  # water
-
-	color_choose = {'Clean': ['#1d4a9f', '#84a7e9'],
-					'Transition': ['#4a9f1d', '#a7e984'],
-					'Event': ['#9f1d4a', '#e984a7']}
-
-	paired = [plt.get_cmap('Paired')(i) for i in range(4)]
-
-	@staticmethod
-	def getColor(num: int = 6, cmap: str = 'jet_r'):
-		category_colors = plt.colormaps[cmap](np.linspace(0.1, 0.9, num))
-		return [plc.to_hex(category_colors[i]) for i in range(num)]
-
-	@staticmethod
-	def palplot(*args, **kwargs):
-		sns.palplot(*args, **kwargs)
-
-	@staticmethod
-	def adjust_opacity(colors: str | list[str], alpha: float):
-		if isinstance(colors, str):
-			colors = [colors]
-
-		adjusted_colors = []
-		for color in colors:
-			# 將顏色轉換為RGB表示
-			r, g, b = int(color[1:3], 16), int(color[3:5], 16), int(color[5:7], 16)
-			# 調整透明度
-			r_new = int(alpha * r + (1 - alpha) * 255)
-			g_new = int(alpha * g + (1 - alpha) * 255)
-			b_new = int(alpha * b + (1 - alpha) * 255)
-			# 轉換為新的色碼
-			new_color = '#{:02X}{:02X}{:02X}'.format(r_new, g_new, b_new)
-			adjusted_colors.append(new_color)
-		return adjusted_colors
-
-	@staticmethod
-	def color_maker(obj, cmap='Blues'):
-		colors = np.nan_to_num(obj, nan=0)
-		scalar_map = plt.cm.ScalarMappable(cmap=colormaps[cmap])  # create a scalar map for the colorbar
-		scalar_map.set_array(colors)
-		return scalar_map, colors
+    color_cycle = cycler(color=['b', 'g', 'r', 'c', 'm', 'y', 'k'])
+
+    linecolor = [{'line': '#1a56db', 'edge': '#0F50A6', 'face': '#5983D9'},
+                 {'line': '#046c4e', 'edge': '#1B591F', 'face': '#538C4A'},
+                 {'line': '#c81e1e', 'edge': '#f05252', 'face': '#f98080'}]
+
+    # colors = ['#FF3333', '#33FF33', '#FFFF33', '#5555FF', '#B94FFF', '#AAAAAA', '#748690'] # the last one is "unknown"
+
+    colors1 = ['#A65E58', '#A5BF6B', '#F2BF5E', '#3F83BF', '#B777C2', '#D1CFCB']
+    colors2 = ['#A65E58', '#A5BF6B', '#F2BF5E', '#3F83BF', '#B777C2', '#D1CFCB', '#96c8e6']
+    colors3 = ['#A65E58', '#A5BF6B', '#a6710d', '#F2BF5E', '#3F83BF', '#B777C2', '#D1CFCB', '#96c8e6']  # POC SOC
+
+    colors_mutiWater = ['#A65E58', '#c18e8a', '#A5BF6B', '#c5d6a0', '#F2BF5E', '#3F83BF', '#c089ca', '#d3acda',
+                        '#D1CFCB']
+    colors_mutiWater2 = ['#A65E58', '#96c8e6', '#A5BF6B', '#96c8e6', '#F2BF5E', '#3F83BF', '#c089ca', '#96c8e6',
+                         '#D1CFCB']  # water
+
+    color_choose = {'Clean': ['#1d4a9f', '#84a7e9'],
+                    'Transition': ['#4a9f1d', '#a7e984'],
+                    'Event': ['#9f1d4a', '#e984a7']}
+
+    paired = [plt.get_cmap('Paired')(i) for i in range(4)]
+
+    @staticmethod
+    def getColor(num: int = 6, cmap: str = 'jet_r'):
+        category_colors = plt.colormaps[cmap](np.linspace(0.1, 0.9, num))
+        return [plc.to_hex(category_colors[i]) for i in range(num)]
+
+    @staticmethod
+    def palplot(*args, **kwargs):
+        sns.palplot(*args, **kwargs)
+
+    @staticmethod
+    def adjust_opacity(colors: str | list[str], alpha: float):
+        if isinstance(colors, str):
+            colors = [colors]
+
+        adjusted_colors = []
+        for color in colors:
+            # 將顏色轉換為RGB表示
+            r, g, b = int(color[1:3], 16), int(color[3:5], 16), int(color[5:7], 16)
+            # 調整透明度
+            r_new = int(alpha * r + (1 - alpha) * 255)
+            g_new = int(alpha * g + (1 - alpha) * 255)
+            b_new = int(alpha * b + (1 - alpha) * 255)
+            # 轉換為新的色碼
+            new_color = '#{:02X}{:02X}{:02X}'.format(r_new, g_new, b_new)
+            adjusted_colors.append(new_color)
+        return adjusted_colors
+
+    @staticmethod
+    def color_maker(obj, cmap='Blues'):
+        colors = np.nan_to_num(obj, nan=0)
+        scalar_map = plt.cm.ScalarMappable(cmap=colormaps[cmap])  # create a scalar map for the colorbar
+        scalar_map.set_array(colors)
+        return scalar_map, colors
 
 
 if __name__ == '__main__':
-	Color.palplot(Color.colors2)
+    Color.palplot(Color.colors2)

AeroViz/plot/utils/_unit.py

@@ -5,51 +5,51 @@ __all__ = ['Unit']
 
 
 class Unit:
-	file_path = Path(__file__).parent / 'units.json'
-	data = None
-
-	def __new__(cls, unit: str):
-		cls.data = cls.load_jsonfile()
-		try:
-			value = cls.data[unit]
-			return r'${}$'.format(value.replace(' ', r'\ '))
-		except KeyError:
-			print(f"Attribute '{unit}' not found. Using default value.")
-			return r'${}$'.format(unit.replace(' ', r'\ ')) if unit is not None else 'None'
-
-	@classmethod
-	def load_jsonfile(cls):
-		""" 讀取 JSON 檔中數據并將其變成屬性 """
-		try:
-			with open(cls.file_path, 'r', encoding='utf-8') as f:
-				return json.load(f)
-
-		except FileNotFoundError:
-			print(f"JSON file '{cls.file_path}' not found.")
-		except json.JSONDecodeError:
-			print(f"Invalid JSON format in '{cls.file_path}'.")
-
-	@classmethod
-	def update_jsonfile(cls, key, value):
-		""" 更新JSON檔 """
-		with open(cls.file_path, 'r', encoding='utf-8') as f:
-			old_data = json.load(f)
-
-		old_data[key] = value
-
-		with open(cls.file_path, 'w', encoding='utf-8') as f:
-			json.dump(old_data, f, indent=4)
-
-	@classmethod
-	def del_jsonfile(cls, key):
-		""" 更新JSON檔 """
-		with open(cls.file_path, 'r', encoding='utf-8') as f:
-			old_data = json.load(f)
-
-		if key in old_data:
-			del old_data[key]
-
-			with open(cls.file_path, 'w', encoding='utf-8') as f:
-				json.dump(old_data, f, indent=4)
-		else:
-			print(f"Key '{key}' not found.")
+    file_path = Path(__file__).parent / 'units.json'
+    data = None
+
+    def __new__(cls, unit: str):
+        cls.data = cls.load_jsonfile()
+        try:
+            value = cls.data[unit]
+            return r'${}$'.format(value.replace(' ', r'\ '))
+        except KeyError:
+            print(f"Attribute '{unit}' not found. Using default value.")
+            return r'${}$'.format(unit.replace(' ', r'\ ')) if unit is not None else 'None'
+
+    @classmethod
+    def load_jsonfile(cls):
+        """ 讀取 JSON 檔中數據并將其變成屬性 """
+        try:
+            with open(cls.file_path, 'r', encoding='utf-8') as f:
+                return json.load(f)
+
+        except FileNotFoundError:
+            print(f"JSON file '{cls.file_path}' not found.")
+        except json.JSONDecodeError:
+            print(f"Invalid JSON format in '{cls.file_path}'.")
+
+    @classmethod
+    def update_jsonfile(cls, key, value):
+        """ 更新JSON檔 """
+        with open(cls.file_path, 'r', encoding='utf-8') as f:
+            old_data = json.load(f)
+
+        old_data[key] = value
+
+        with open(cls.file_path, 'w', encoding='utf-8') as f:
+            json.dump(old_data, f, indent=4)
+
+    @classmethod
+    def del_jsonfile(cls, key):
+        """ 更新JSON檔 """
+        with open(cls.file_path, 'r', encoding='utf-8') as f:
+            old_data = json.load(f)
+
+        if key in old_data:
+            del old_data[key]
+
+            with open(cls.file_path, 'w', encoding='utf-8') as f:
+                json.dump(old_data, f, indent=4)
+        else:
+            print(f"Key '{key}' not found.")