oafuncs 0.0.81__py2.py3-none-any.whl → 0.0.83__py2.py3-none-any.whl

oafuncs/oa_file.py

@@ -1,38 +1,39 @@
 #!/usr/bin/env python
 # coding=utf-8
-'''
+"""
 Author: Liu Kun && 16031215@qq.com
 Date: 2024-09-17 15:07:13
 LastEditors: Liu Kun && 16031215@qq.com
-LastEditTime: 2024-12-02 10:33:19
+LastEditTime: 2024-12-13 16:28:56
 FilePath: \\Python\\My_Funcs\\OAFuncs\\oafuncs\\oa_file.py
 Description:
 EditPlatform: vscode
 ComputerInfo: XPS 15 9510
 SystemInfo: Windows 11
 Python Version: 3.11
-'''
-
+"""
 
 import glob
 import os
 import re
 import shutil
 
-__all__ = ['find_file', 'link_file', 'copy_file', 'rename_file', 'make_folder', 'clear_folder', 'remove_empty_folders', 'remove', 'file_size']
+__all__ = ["find_file", "link_file", "copy_file", "rename_file", "make_folder", "clear_folder", "remove_empty_folder", "remove", "file_size"]
 
 
-def find_file(parent_path, fname, mode='path'):
-    '''
-    description: 
+# ** 查找文件，支持通配符
+def find_file(parent_path, fname, mode="path"):
+    """
+    description:
     param {*} parent_path: The parent path where the files are located
     param {*} fname: The file name pattern to search for
     param {*} mode: 'path' to return the full path of the files, 'file' to return only the file names
     return {*} A list of file paths or file names if files are found, None otherwise
-    '''
+    """
+
     def natural_sort_key(s):
         """生成一个用于自然排序的键"""
-        return [int(text) if text.isdigit() else text.lower() for text in re.split('([0-9]+)', s)]
+        return [int(text) if text.isdigit() else text.lower() for text in re.split("([0-9]+)", s)]
 
     # 将parent_path和fname结合成完整的搜索路径
     search_pattern = os.path.join(str(parent_path), fname)
@@ -48,7 +49,7 @@ def find_file(parent_path, fname, mode='path'):
     matched_files = sorted(matched_files, key=natural_sort_key)
 
     # 根据mode参数决定返回的内容
-    if mode == 'file':
+    if mode == "file":
         # 只返回文件名
         result = [os.path.basename(file) for file in matched_files]
     else:  # 默认为'path'
@@ -58,8 +59,9 @@ def find_file(parent_path, fname, mode='path'):
     return result
 
 
+# ** 创建符号链接，支持通配符
 def link_file(src_pattern, dst):
-    '''
+    """
     # 描述：创建符号链接，支持通配符
     # 使用示例
     # link_file(r'/data/hejx/liukun/era5/*', r'/data/hejx/liukun/Test/')
@@ -67,12 +69,12 @@ def link_file(src_pattern, dst):
     # link_file(r'/data/hejx/liukun/era5/py.o*', r'/data/hejx/liukun/Test')
     param      {*} src_pattern # 源文件或目录
     param      {*} dst # 目标文件或目录
-    '''
+    """
     src_pattern = str(src_pattern)
     # 使用glob.glob来处理可能包含通配符的src
     src_files = glob.glob(src_pattern)
     if not src_files:
-        raise FileNotFoundError('文件不存在: {}'.format(src_pattern))
+        raise FileNotFoundError("文件不存在: {}".format(src_pattern))
 
     # 判断dst是路径还是包含文件名的路径
     if os.path.isdir(dst):
@@ -84,7 +86,7 @@ def link_file(src_pattern, dst):
             if os.path.exists(dst_file):
                 os.remove(dst_file)
             os.symlink(src_file, dst_file)
-            print(f'创建符号链接: {src_file} -> {dst_file}')
+            print(f"创建符号链接: {src_file} -> {dst_file}")
     else:
         # 如果dst包含文件名，则创建链接后重命名
         dst_dir = os.path.dirname(dst)
@@ -95,23 +97,24 @@ def link_file(src_pattern, dst):
         if os.path.exists(dst_file):
             os.remove(dst_file)
         os.symlink(src_file, dst_file)
-        print(f'创建符号链接并重命名: {src_file} -> {dst_file}')
+        print(f"创建符号链接并重命名: {src_file} -> {dst_file}")
 
 
+# ** 复制文件或目录，支持通配符
 def copy_file(src_pattern, dst):
-    '''
+    """
     # 描述：复制文件或目录，支持通配符
     # 使用示例
     # copy_file(r'/data/hejx/liukun/era5/py.o*', r'/data/hejx/liukun/Test/py.o')
     # copy_file(r'/data/hejx/liukun/era5/py.o*', r'/data/hejx/liukun/Test')
     param      {*} src_pattern # 源文件或目录
     param      {*} dst # 目标文件或目录
-    '''
+    """
     src_pattern = str(src_pattern)
     # 使用glob.glob来处理可能包含通配符的src
     src_files = glob.glob(src_pattern)
     if not src_files:
-        raise FileNotFoundError('文件不存在: {}'.format(src_pattern))
+        raise FileNotFoundError("文件不存在: {}".format(src_pattern))
 
     # 判断dst是路径还是包含文件名的路径
     if os.path.isdir(dst):
@@ -129,7 +132,7 @@ def copy_file(src_pattern, dst):
                 shutil.copytree(src_file, dst_file, symlinks=True)
             else:
                 shutil.copy2(src_file, dst_file)
-            print(f'复制文件或目录: {src_file} -> {dst_file}')
+            print(f"复制文件或目录: {src_file} -> {dst_file}")
     else:
         # 如果dst包含文件名，则复制后重命名
         dst_dir = os.path.dirname(dst)
@@ -146,11 +149,12 @@ def copy_file(src_pattern, dst):
             shutil.copytree(src_file, dst_file, symlinks=True)
         else:
             shutil.copy2(src_file, dst_file)
-        print(f'复制文件或目录并重命名: {src_file} -> {dst_file}')
+        print(f"复制文件或目录并重命名: {src_file} -> {dst_file}")
 
 
+# ** 重命名文件，支持通配符
 def rename_file(directory, old_str, new_str):
-    '''
+    """
     # 描述：重命名目录下的文件，支持通配符
     # 使用示例
     directory_path = r"E:\windfarm\CROCO_FILES"
@@ -160,7 +164,7 @@ def rename_file(directory, old_str, new_str):
     param      {*} directory # 目录
     param      {*} old_str # 要替换的字符串
     param      {*} new_str # 新字符串
-    '''
+    """
     # 获取目录下的所有文件
     files = os.listdir(directory)
 
@@ -186,8 +190,8 @@ def rename_file(directory, old_str, new_str):
 
 
 # ** 创建子文件夹（可选清空）
-def make_folder(rootpath: str, folder_name: str, clear=0) -> str:
-    '''
+def make_folder(rootpath=None, folder_name=None, clear=0) -> str:
+    """
     # 描述：创建子文件夹（可选清空）
     # 使用示例
     rootpath = r'E:\Data\2024\09\17'
@@ -195,23 +199,24 @@ def make_folder(rootpath: str, folder_name: str, clear=0) -> str:
     newpath = make_folder(rootpath, folder_name, clear=1)
     param       {*} rootpath # 根目录
     param       {*} folder_name # 文件夹名称
-    '''
+    """
+    if rootpath is None:
+        rootpath = os.getcwd()
     folder_path = os.path.join(str(rootpath), str(folder_name))
     if clear:
         shutil.rmtree(folder_path, ignore_errors=True)
     os.makedirs(folder_path, exist_ok=True)
     return folder_path
 
-# ** 清空文件夹
-
 
+# ** 清空文件夹
 def clear_folder(folder_path):
-    '''
+    """
     # 描述：清空文件夹
     # 使用示例
     clear_folder(r'E:\Data\2024\09\17\var1')
     param        {*} folder_path # 文件夹路径
-    '''
+    """
     folder_path = str(folder_path)
     if os.path.exists(folder_path):
         try:
@@ -223,21 +228,21 @@ def clear_folder(folder_path):
                     os.unlink(file_path)  # 删除文件或链接
                 elif os.path.isdir(file_path):
                     shutil.rmtree(file_path)  # 删除子文件夹
-            print(f'成功清空文件夹: {folder_path}')
+            print(f"成功清空文件夹: {folder_path}")
         except Exception as e:
-            print(f'清空文件夹失败: {folder_path}')
+            print(f"清空文件夹失败: {folder_path}")
             print(e)
 
 
 # ** 清理空文件夹
-def remove_empty_folders(path, print_info=1):
-    '''
+def remove_empty_folder(path, print_info=1):
+    """
     # 描述：清理空文件夹
     # 使用示例
-    remove_empty_folders(r'E:\Data\2024\09\17', print_info=1)
+    remove_empty_folder(r'E:\Data\2024\09\17', print_info=1)
     param        {*} path # 文件夹路径
     param        {*} print_info # 是否打印信息
-    '''
+    """
     path = str(path)
     # 遍历当前目录下的所有文件夹和文件
     for root, dirs, files in os.walk(path, topdown=False):
@@ -263,14 +268,14 @@ def remove_empty_folders(path, print_info=1):
 
 # ** 删除相关文件，可使用通配符
 def remove(pattern):
-    '''
+    """
     # 描述：删除相关文件，可使用通配符
     remove(r'E:\Code\Python\Model\WRF\Radar2\bzip2-radar-0*')
     # or
     os.chdir(r'E:\Code\Python\Model\WRF\Radar2')
     remove('bzip2-radar-0*')
     param        {*} pattern # 文件路径或通配符
-    '''
+    """
     # 使用glob.glob来获取所有匹配的文件
     # 可以使用通配符*来匹配所有文件
     pattern = str(pattern)
@@ -279,16 +284,17 @@ def remove(pattern):
         if os.path.exists(file_path):
             try:
                 shutil.rmtree(file_path)
-                print(f'成功删除文件: {file_path}')
+                print(f"成功删除文件: {file_path}")
             except Exception as e:
-                print(f'删除文件失败: {file_path}')
+                print(f"删除文件失败: {file_path}")
                 print(e)
         else:
-            print(f'文件不存在: {file_path}')
+            print(f"文件不存在: {file_path}")
 
 
-def file_size(file_path, unit='KB'):
-    '''
+# ** 获取文件大小
+def file_size(file_path, unit="KB"):
+    """
     Description: 获取文件大小
 
     Args:
@@ -297,7 +303,7 @@ def file_size(file_path, unit='KB'):
 
     Returns:
     文件大小（单位：PB、TB、GB、MB、KB）
-    '''
+    """
     # 检查文件是否存在
     if not os.path.exists(file_path):
         return "文件不存在"
@@ -306,13 +312,7 @@ def file_size(file_path, unit='KB'):
     file_size = os.path.getsize(file_path)
 
     # 单位转换字典
-    unit_dict = {
-        'PB': 1024**5,
-        'TB': 1024**4,
-        'GB': 1024**3,
-        'MB': 1024**2,
-        'KB': 1024
-    }
+    unit_dict = {"PB": 1024**5, "TB": 1024**4, "GB": 1024**3, "MB": 1024**2, "KB": 1024}
 
     # 检查传入的单位是否合法
     if unit not in unit_dict:
@@ -324,9 +324,22 @@ def file_size(file_path, unit='KB'):
     return converted_size
 
 
-if __name__ == '__main__':
+# ** 计算文件夹下指定相关文件的平均大小
+def mean_size(parent_path,fname):
+    flist = find_file(parent_path, fname)
+    if flist:
+        size_list = [file_size(f) for f in flist if file_size(f) != 0]
+    if size_list:
+        min_size, max_size = min(size_list), max(size_list)
+        mean_size = sum(size_list) / len(size_list)
+    else:
+        mean_size, min_size, max_size = 0, 0, 0
+    return mean_size, min_size, max_size
+
+
+if __name__ == "__main__":
     # newpath = make_folder('D:/Data/2024/09/17/', 'var1', clear=1)
     # print(newpath)
     pass
 
-    remove(r'I:\Delete\test\*')
+    remove(r"I:\Delete\test\*")

oafuncs/oa_help.py

@@ -1,39 +1,42 @@
 #!/usr/bin/env python
 # coding=utf-8
-'''
+"""
 Author: Liu Kun && 16031215@qq.com
 Date: 2024-10-06 19:25:29
 LastEditors: Liu Kun && 16031215@qq.com
-LastEditTime: 2024-11-21 12:59:18
+LastEditTime: 2024-12-26 10:43:40
 FilePath: \\Python\\My_Funcs\\OAFuncs\\oafuncs\\oa_help.py
-Description:  
+Description:
 EditPlatform: vscode
 ComputerInfo: XPS 15 9510
 SystemInfo: Windows 11
-Python Version: 3.11
-'''
+Python Version: 3.12
+"""
 
 import oafuncs
 
+__all__ = ["query", "use"]
+
+
 def query():
-    '''
-    description: 查看OAFuncs模块的函数列表
+    """
+    description: 查看oafuncs模块的函数列表
     example: query()
-    '''
+    """
     funcs = [func for func in dir(oafuncs) if callable(getattr(oafuncs, func))]
-    print('函数数量：')
+    print("函数数量：")
     print(len(funcs))
-    print('函数列表：')
+    print("函数列表：")
     print(funcs)
 
 
-def use(func='get_var'):
-    '''
+def use(func="get_var"):
+    """
     description: 查看函数的模块全路径和函数提示
     param {func} : 函数名
     example: use('get_var')
-    '''
-    print('模块全路径：')
-    print(getattr(oafuncs, func).__module__+'.'+func)
-    print('函数提示：')
+    """
+    print("模块全路径：")
+    print(getattr(oafuncs, func).__module__ + "." + func)
+    print("函数提示：")
     print(getattr(oafuncs, func).__doc__)

oafuncs/oa_nc.py

@@ -19,7 +19,7 @@ import netCDF4 as nc
 import numpy as np
 import xarray as xr
 
-__all__ = ["get_var", "extract5nc", "write2nc", "merge5nc", "modify_var_value", "modify_var_attr", "rename_var_or_dim", "check_ncfile", "longitude_change", "nc_isel"]
+__all__ = ["get_var", "extract5nc", "write2nc", "merge5nc", "modify_var_value", "modify_var_attr", "rename_var_or_dim", "check_ncfile", "convert_longitude", "nc_isel"]
 
 
 def get_var(file, *vars):
@@ -79,21 +79,32 @@ def _numpy_to_nc_type(numpy_type):
         "uint32": "u4",
         "uint64": "u8",
     }
-    return numpy_to_nc.get(str(numpy_type), "f4")  # 默认使用 'float32'
+    # 确保传入的是字符串类型，如果不是，则转换为字符串
+    numpy_type_str = str(numpy_type) if not isinstance(numpy_type, str) else numpy_type
+    return numpy_to_nc.get(numpy_type_str, "f4")  # 默认使用 'float32'
 
 
 def _calculate_scale_and_offset(data, n=16):
-    data_min, data_max = np.nanmin(data), np.nanmax(data)
-    scale_factor = (data_max - data_min) / (2 ** n - 1)
+    if not isinstance(data, np.ndarray):
+        raise ValueError("Input data must be a NumPy array.")
+
+    # 使用 nan_to_num 来避免 NaN 值对 min 和 max 的影响
+    data_min = np.nanmin(data)
+    data_max = np.nanmax(data)
+
+    if np.isnan(data_min) or np.isnan(data_max):
+        raise ValueError("Input data contains NaN values, which are not allowed.")
+
+    scale_factor = (data_max - data_min) / (2**n - 1)
     add_offset = data_min + 2 ** (n - 1) * scale_factor
-    # S = Q * scale_factor + add_offset
+
     return scale_factor, add_offset
 
 
-def write2nc(file, data, varname=None, coords=None, mode='w', scale_offset_switch=True, compile_switch=True):
+def write2nc(file, data, varname=None, coords=None, mode="w", scale_offset_switch=True, compile_switch=True):
     """
     description: 写入数据到nc文件
-    
+
     参数：
     file: 文件路径
     data: 数据
@@ -102,115 +113,66 @@ def write2nc(file, data, varname=None, coords=None, mode='w', scale_offset_switc
     mode: 写入模式，'w'为写入，'a'为追加
     scale_offset_switch: 是否使用scale_factor和add_offset，默认为True
     compile_switch: 是否使用压缩参数，默认为True
-    
+
     example: write2nc(r'test.nc', data, 'data', {'time': np.linspace(0, 120, 100), 'lev': np.linspace(0, 120, 50)}, 'a')
     """
-    # 判断mode是写入还是追加
-    if mode == "w":
-        if os.path.exists(file):
-            os.remove(file)
-            print("Warning: File already exists. Deleting it.")
-    elif mode == "a":
-        if not os.path.exists(file):
-            print("Warning: File doesn't exist. Creating a new file.")
-            mode = "w"
-    
-    complete = False
-    if varname is None and coords is None:
-        try:
-            data.to_netcdf(file)
-            complete = True
-            # 不能在这里return
-        except AttributeError:
-            raise ValueError("If varname and coords are None, data must be a DataArray.")
-    
-    if complete:
-        return
-    
-    kwargs = {'zlib': True, 'complevel': 4} # 压缩参数
-    # kwargs = {"compression": 'zlib', "complevel": 4}  # 压缩参数
+    # 设置压缩参数
+    kwargs = {"zlib": True, "complevel": 4} if compile_switch else {}
+
+    # 检查文件存在性并根据模式决定操作
+    if mode == "w" and os.path.exists(file):
+        os.remove(file)
+    elif mode == "a" and not os.path.exists(file):
+        mode = "w"
 
     # 打开 NetCDF 文件
     with nc.Dataset(file, mode, format="NETCDF4") as ncfile:
-        # 处理坐标
+        # 如果 data 是 DataArray 并且没有提供 varname 和 coords
+        if varname is None and coords is None and isinstance(data, xr.DataArray):
+            data.to_netcdf(file, mode=mode)
+            return
+
+        # 添加坐标
         for dim, coord_data in coords.items():
-            add_coords = True
-            # 判断坐标是否存在，若存在，则替换/报错
-            if ncfile.dimensions:
-                # 返回字典，字典、列表、元组若为空，都表示False
-                if dim in ncfile.dimensions:
-                    # del nc.dimensions[dim]
-                    if len(coord_data) != len(ncfile.dimensions[dim]):
-                        raise ValueError("Length of coordinate does not match the dimension length.")
-                    else:
-                        add_coords = False
-                        print(f"Warning: Coordinate '{dim}' already exists. Replacing it.")
-                        ncfile.variables[dim][:] = np.array(coord_data)
-            if add_coords:
-                # 创建新坐标
-                ncfile.createDimension(dim, len(coord_data))
-                if compile_switch:
-                    ncfile.createVariable(dim, _numpy_to_nc_type(coord_data.dtype), (dim,), **kwargs)
+            if dim in ncfile.dimensions:
+                if len(coord_data) != len(ncfile.dimensions[dim]):
+                    raise ValueError(f"Length of coordinate '{dim}' does not match the dimension length.")
                 else:
-                    ncfile.createVariable(dim, _numpy_to_nc_type(coord_data.dtype), (dim,))
-                ncfile.variables[dim][:] = np.array(coord_data)
-            
-            if isinstance(coord_data, xr.DataArray):
-                current_var = ncfile.variables[dim]
-                if coord_data.attrs:
-                    for key, value in coord_data.attrs.items():
-                        current_var.setncattr(key, value)
-
-        # 判断变量是否存在，若存在，则删除原变量
-        add_var = True
+                    ncfile.variables[dim][:] = np.array(coord_data)
+            else:
+                ncfile.createDimension(dim, len(coord_data))
+                var = ncfile.createVariable(dim, _numpy_to_nc_type(coord_data.dtype), (dim,), **kwargs)
+                var[:] = np.array(coord_data)
+
+                # 如果坐标数据有属性，则添加到 NetCDF 变量
+                if isinstance(coord_data, xr.DataArray) and coord_data.attrs:
+                    for attr_name, attr_value in coord_data.attrs.items():
+                        var.setncattr(attr_name, attr_value)
+
+        # 添加或更新变量
         if varname in ncfile.variables:
-            print(f"Warning: Variable '{varname}' already exists.")
             if data.shape != ncfile.variables[varname].shape:
-                raise ValueError("Shape of data does not match the variable shape.")
-            else:
-                # 写入数据
-                ncfile.variables[varname][:] = np.array(data)
-                add_var = False
-                print(f"Warning: Variable '{varname}' already exists. Replacing it.")
-
-        if add_var:
-            # 创建变量及其维度
-            dim_names = tuple(coords.keys())  # 使用coords传入的维度名称
+                raise ValueError(f"Shape of data does not match the variable shape for '{varname}'.")
+            ncfile.variables[varname][:] = np.array(data)
+        else:
+            # 创建变量
+            dim_names = tuple(coords.keys())
             if scale_offset_switch:
                 scale_factor, add_offset = _calculate_scale_and_offset(np.array(data))
-                _FillValue = -32767
-                missing_value = -32767
-                dtype = 'i2' # short类型
+                dtype = "i2"
+                var = ncfile.createVariable(varname, dtype, dim_names, fill_value=-32767, **kwargs)
+                var.setncattr("scale_factor", scale_factor)
+                var.setncattr("add_offset", add_offset)
             else:
                 dtype = _numpy_to_nc_type(data.dtype)
-            
-            if compile_switch:
-                ncfile.createVariable(varname, dtype, dim_names, **kwargs)
-            else:
-                ncfile.createVariable(varname, dtype, dim_names)
-            
-            if scale_offset_switch: # 需要在写入数据之前设置scale_factor和add_offset
-                ncfile.variables[varname].setncattr('scale_factor', scale_factor)
-                ncfile.variables[varname].setncattr('add_offset', add_offset)
-                ncfile.variables[varname].setncattr('_FillValue', _FillValue)
-                ncfile.variables[varname].setncattr('missing_value', missing_value)
-                
-            # ncfile.createVariable('data', 'f4', ('time','lev'))
-
-            # 写入数据
-            ncfile.variables[varname][:] = np.array(data)
+                var = ncfile.createVariable(varname, dtype, dim_names, **kwargs)
+            var[:] = np.array(data)
 
-        # 判断维度是否匹配
-        if len(data.shape) != len(coords):
-            raise ValueError("Number of dimensions does not match the data shape.")
-    # 判断data是否带有属性信息，如果有，写入属性信息
-    if isinstance(data, xr.DataArray):
-        current_var = ncfile.variables[varname]
-        if data.attrs:
+        # 添加属性
+        if isinstance(data, xr.DataArray) and data.attrs:
             for key, value in data.attrs.items():
-                if key in ["scale_factor", "add_offset", "_FillValue", "missing_value"] and scale_offset_switch:
-                    continue
-                current_var.setncattr(key, value)
+                if key not in ["scale_factor", "add_offset", "_FillValue", "missing_value"] or not scale_offset_switch:
+                    var.setncattr(key, value)
 
 
 def merge5nc(file_list, var_name=None, dim_name=None, target_filename=None):
@@ -230,6 +192,9 @@ def merge5nc(file_list, var_name=None, dim_name=None, target_filename=None):
     merge5nc(file_list, var_name=['data1', 'data2'], dim_name='time', target_filename='merged.nc')
     merge5nc(file_list, var_name=None, dim_name='time', target_filename='merged.nc')
     """
+    if isinstance(file_list, str):
+        file_list = [file_list]
+    
     # 初始化变量名列表
     var_names = None
 
@@ -398,25 +363,28 @@ def check_ncfile(ncfile, if_delete=False):
         return False
 
 
-def longitude_change(ds, lon_name="longitude", to_which="180"):
+def convert_longitude(ds, lon_name="longitude", convert=180):
     """
-    将经度转换为 -180 到 180 之间
+    将经度数组转换为指定的范围。
 
     参数：
-    lon (numpy.ndarray): 经度数组
+    ds (xarray.Dataset): 包含经度数据的xarray数据集。
+    lon_name (str): 经度变量的名称，默认为"longitude"。
+    convert (int): 转换目标范围，可以是180或360，默认为180。
 
     返回值：
-    numpy.ndarray: 转换后的经度数组
+    xarray.Dataset: 经度转换后的xarray数据集。
     """
-    # return (lon + 180) % 360 - 180
-    # ds = ds.assign_coords(longitude=(((ds.longitude + 180) % 360) - 180)).sortby("longitude")
-    if to_which == "180":
-        # ds = ds.assign_coords(**{lon_name: (((ds[lon_name] + 180) % 360) - 180)}).sortby(lon_name)
-        ds = ds.assign_coords(**{lon_name: (ds[lon_name] + 180) % 360 - 180}).sortby(lon_name)
-    elif to_which == "360":
-        # -180 to 180 to 0 to 360
-        ds = ds.assign_coords(**{lon_name: (ds[lon_name] + 360) % 360}).sortby(lon_name)
-    return ds
+    to_which = int(convert)
+    if to_which not in [180, 360]:
+        raise ValueError("to_which must be '180' or '360'")
+
+    if to_which == 180:
+        ds = ds.assign_coords({lon_name: (ds[lon_name] + 180) % 360 - 180})
+    elif to_which == 360:
+        ds = ds.assign_coords({lon_name: (ds[lon_name] + 360) % 360})
+
+    return ds.sortby(lon_name)
 
 
 def nc_isel(ncfile, dim_name, slice_list):