oafuncs 0.0.98.37__py3-none-any.whl → 0.0.98.39__py3-none-any.whl

oafuncs/_script/parallel.py

@@ -155,7 +155,8 @@ class ParallelExecutor:
                 # 实时更新进度条
                 progress_bar.update(1)
                 progress_bar.refresh()
-        print('\n') # 结束进度条输出
+        # 完成后换行
+        print()
         return results
 
     def _chunked_execution(self, func: Callable, params: List[Tuple], chunk_size: int) -> List[Any]:
@@ -190,7 +191,8 @@ class ParallelExecutor:
                 # 更新分块进度
                 progress_bar.update(1)
                 progress_bar.refresh()
-        print('\n')
+        # 完成后换行
+        print()
         return results
 
     @staticmethod

oafuncs/oa_cmap.py

@@ -313,6 +313,7 @@ def random_color():
     b = random.randint(0, 255)
     return f"#{r:02x}{g:02x}{b:02x}"
 
+
 if __name__ == "__main__":
     # ** 测试自制cmap
     colors = ["#C2B7F3", "#B3BBF2", "#B0CBF1", "#ACDCF0", "#A8EEED"]

oafuncs/oa_draw.py

@@ -6,7 +6,7 @@ import matplotlib.pyplot as plt
 import numpy as np
 from rich import print
 
-__all__ = ["fig_minus", "gif", "movie", "setup_map", "MidpointNormalize"]
+__all__ = ["fig_minus", "gif", "movie", "setup_map", "ticks_symmetric"]
 
 warnings.filterwarnings("ignore")
 
@@ -314,57 +314,47 @@ def setup_map(
     return axes
 
 
-class MidpointNormalize(mpl.colors.Normalize):
-    """Custom normalization class to center a specific value.
-
-    Args:
-        vmin (float, optional): Minimum data value. Defaults to None.
-        vmax (float, optional): Maximum data value. Defaults to None.
-        vcenter (float, optional): Center value for normalization. Defaults to 0.
-        clip (bool, optional): Whether to clip data outside the range. Defaults to False.
-
-    Example:
-        >>> norm = MidpointNormalize(vmin=-2, vmax=1, vcenter=0)
+def ticks_symmetric(vmin: float, vcenter: float, vmax: float, num: int = 7) -> np.ndarray:
     """
-
-    def __init__(self, vmin: float = None, vmax: float = None, vcenter: float = 0, clip: bool = False) -> None:
-        self.vcenter = vcenter
-        super().__init__(vmin, vmax, clip)
-
-    def __call__(self, value: np.ndarray, clip: bool = None) -> np.ma.MaskedArray:
-        # Use the clip parameter from initialization if not provided
-        if clip is None:
-            clip = self.clip
-
-        x, y = [self.vmin, self.vcenter, self.vmax], [0, 0.5, 1.0]
-        result = np.interp(value, x, y)
-
-        # Apply clipping if requested
-        if clip:
-            result = np.clip(result, 0, 1)
-
-        return np.ma.masked_array(result)
+    生成以指定中心点对称分布的刻度值
     
-    def ticks(self, num_ticks: int = 7) -> np.ndarray:
-        """Generate ticks for the normalization range, centered around vcenter."""
-        if self.vmin is None or self.vmax is None:
-            raise ValueError("vmin and vmax must be set to generate ticks.")
-
-        if num_ticks % 2 == 0:
-            num_ticks += 1
-
-        num_points_side = (num_ticks - 1) // 2 + 1
-
-        negative_ticks = np.linspace(self.vmin, self.vcenter, num_points_side)[:-1]
-        positive_ticks = np.linspace(self.vcenter, self.vmax, num_points_side)[1:]
-
-        ticks = np.concatenate([negative_ticks, [self.vcenter], positive_ticks])
-
-        return ticks
-
-    def inverse(self, value: np.ndarray) -> np.ndarray:
-        y, x = [self.vmin, self.vcenter, self.vmax], [0, 0.5, 1]
-        return np.interp(value, x, y)
+    参数:
+        vmin (float): 最小值
+        vcenter (float): 中心值
+        vmax (float): 最大值
+        num (int, optional): 期望的刻度数量（必须是奇数）。默认为7
+    
+    返回:
+        np.ndarray: 对称分布的刻度值数组
+    
+    异常:
+        ValueError: 如果输入值无效
+    
+    示例:
+        >>> ticks_symmetric(vmin=-10, vcenter=0, vmax=10, num=5)
+        array([-10.,  -5.,   0.,   5.,  10.])
+    """
+    # 验证输入参数
+    if vmin >= vcenter:
+        raise ValueError(f"vmin ({vmin}) must be less than vcenter ({vcenter})")
+    if vcenter >= vmax:
+        raise ValueError(f"vcenter ({vcenter}) must be less than vmax ({vmax})")
+    
+    # 确保刻度数量是奇数
+    if num % 2 == 0:
+        num += 1
+    
+    # 计算每侧的点数（包括中心点）
+    side_points = (num - 1) // 2 + 1
+    
+    # 生成左侧刻度（从最小值到中心值）
+    left_ticks = np.linspace(vmin, vcenter, side_points)[:-1]
+    
+    # 生成右侧刻度（从中心值到最大值）
+    right_ticks = np.linspace(vcenter, vmax, side_points)[1:]
+    
+    # 组合所有刻度
+    return np.concatenate([left_ticks, [vcenter], right_ticks])
 
 
 if __name__ == "__main__":

{oafuncs-0.0.98.37.dist-info → oafuncs-0.0.98.39.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: oafuncs
-Version: 0.0.98.37
+Version: 0.0.98.39
 Summary: Oceanic and Atmospheric Functions
 Home-page: https://github.com/Industry-Pays/OAFuncs
 Author: Kun Liu

{oafuncs-0.0.98.37.dist-info → oafuncs-0.0.98.39.dist-info}/RECORD

@@ -1,8 +1,8 @@
 oafuncs/__init__.py,sha256=T_-VtnWWllV3Q91twT5Yt2sUapeA051QbPNnBxmg9nw,1456
-oafuncs/oa_cmap.py,sha256=h9QrOSwbzZJXuS2pjFZkJtck1bq9itPY1natIuIRB3s,13884
+oafuncs/oa_cmap.py,sha256=JwZMJ36uNwiCnzXqEtH2_PpeLtEaRaXP9YeGSl0PJSU,13886
 oafuncs/oa_data.py,sha256=u4H1ZazQf2jmGx3IAiaGxLjQHY9cQEMizRIaTQW4UiE,8075
 oafuncs/oa_date.py,sha256=aU2wVIWXyWoRiSQ9dg8sHvShFTxw86RrgbV3Q6tDjD4,6841
-oafuncs/oa_draw.py,sha256=UnUyOcr6KHtEWCZG0TCnHFW0jrjkAW9XWZoUDcSs6oI,14386
+oafuncs/oa_draw.py,sha256=zxwxmTsS6-s6YFvrVJTr06_O39y0ykVxtOrShgDM4Qg,13754
 oafuncs/oa_file.py,sha256=fLb0gRhq2AiPl-5ASDHMrx6Z267FmhqNcTV7CdCxTdI,16934
 oafuncs/oa_help.py,sha256=0J5VaZX-cB0c090KxgmktQJBc0o00FsY-4wB8l5y00k,4178
 oafuncs/oa_nc.py,sha256=mKNxQ9jPxfRH7xINyrX7tBhitG5gmOKm6Dn7stk5mdw,15279
@@ -16,7 +16,7 @@ oafuncs/_script/email.py,sha256=l5xDgdVj8O5V0J2SwjsHKdUuxOH2jZvwdMO_P0dImHU,2684
 oafuncs/_script/netcdf_merge.py,sha256=tM9ePqLiEsE7eIsNM5XjEYeXwxjYOdNz5ejnEuI7xKw,6066
 oafuncs/_script/netcdf_modify.py,sha256=XDlAEToe_lwfAetkBSENqU5df-wnH7MGuxNTjG1gwHY,4178
 oafuncs/_script/netcdf_write.py,sha256=GvyUyUhzMonzSp3y4pT8ZAfbQrsh5J3dLnmINYJKhuE,21422
-oafuncs/_script/parallel.py,sha256=glEeZEg6HU3q1E6kUF-9k4l__KmIa3KlOglbOUcqubU,10047
+oafuncs/_script/parallel.py,sha256=VMNhK3PNcZrIj-ZxcmAWuU3mIfVsfztsk2Ceqwri4e4,10069
 oafuncs/_script/parallel_bak.py,sha256=2ySmYZ9e_PLhhMocWCCFWCYZD3Gs_mxl0HxEzbIuQvA,8861
 oafuncs/_script/plot_dataset.py,sha256=QrA4vOCzWbAJp3hf5YYzgIRUZdJB5_ugepgyT_YfnaY,16327
 oafuncs/_script/replace_file_content.py,sha256=wIwvaISFNYWG58BLZHZP9ZgbC5OhoZ-cpR3y25U1EUM,5601
@@ -37,8 +37,8 @@ oafuncs/oa_sign/__init__.py,sha256=JSx1fcWpmNhQBvX_Bmq3xysfSkkFMrjbJASxV_V6aqE,1
 oafuncs/oa_sign/meteorological.py,sha256=3MSjy7HTcvz2zsITkjUMr_0Y027Gas1LFE9pk99990k,6110
 oafuncs/oa_sign/ocean.py,sha256=3uYEzaq-27yVy23IQoqy-clhWu1I_fhPFBAQyT-OF4M,5562
 oafuncs/oa_sign/scientific.py,sha256=moIl2MEY4uitbXoD596JmXookXGQtQsS-8_1NBBTx84,4689
-oafuncs-0.0.98.37.dist-info/licenses/LICENSE.txt,sha256=rMtLpVg8sKiSlwClfR9w_Dd_5WubTQgoOzE2PDFxzs4,1074
-oafuncs-0.0.98.37.dist-info/METADATA,sha256=EuBjhFOpWRAcZeADmTIFmg5FUSholH23O3qn5CsCR5U,4326
-oafuncs-0.0.98.37.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-oafuncs-0.0.98.37.dist-info/top_level.txt,sha256=bgC35QkXbN4EmPHEveg_xGIZ5i9NNPYWqtJqaKqTPsQ,8
-oafuncs-0.0.98.37.dist-info/RECORD,,
+oafuncs-0.0.98.39.dist-info/licenses/LICENSE.txt,sha256=rMtLpVg8sKiSlwClfR9w_Dd_5WubTQgoOzE2PDFxzs4,1074
+oafuncs-0.0.98.39.dist-info/METADATA,sha256=2hppaUc6XZv609HGI_ZRS8GRPARn1Q9MI8xgb4JBfhs,4326
+oafuncs-0.0.98.39.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+oafuncs-0.0.98.39.dist-info/top_level.txt,sha256=bgC35QkXbN4EmPHEveg_xGIZ5i9NNPYWqtJqaKqTPsQ,8
+oafuncs-0.0.98.39.dist-info/RECORD,,