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

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"]
+__all__ = ["get_var", "extract5nc", "write2nc", "merge5nc", "modify_var_value", "modify_var_attr", "rename_var_or_dim", "check_ncfile", "longitude_change", "nc_isel"]
 
 
 def get_var(file, *vars):
@@ -38,7 +38,7 @@ def get_var(file, *vars):
     return datas
 
 
-def extract5nc(file, varname):
+def extract5nc(file, varname, only_value=True):
     """
     描述：
     1、提取nc文件中的变量
@@ -47,16 +47,22 @@ def extract5nc(file, varname):
     参数：
     file: 文件路径
     varname: 变量名
+    only_value: 变量和维度是否只保留数值
     example: data, dimdict = extract5nc(file_ecm, 'h')
     """
     ds = xr.open_dataset(file)
     vardata = ds[varname]
+    ds.close()
     dims = vardata.dims
     dimdict = {}
     for dim in dims:
-        dimdict[dim] = ds[dim].values
-    ds.close()
-    return np.array(vardata), dimdict
+        if only_value:
+            dimdict[dim] = vardata[dim].values
+        else:
+            dimdict[dim] = ds[dim]
+    if only_value:
+        vardata = np.array(vardata)
+    return vardata, dimdict
 
 
 def _numpy_to_nc_type(numpy_type):
@@ -76,15 +82,27 @@ def _numpy_to_nc_type(numpy_type):
     return numpy_to_nc.get(str(numpy_type), "f4")  # 默认使用 'float32'
 
 
-def write2nc(file, data, varname, coords, mode):
+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)
+    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):
     """
     description: 写入数据到nc文件
+    
     参数：
     file: 文件路径
     data: 数据
     varname: 变量名
     coords: 坐标，字典，键为维度名称，值为坐标数据
     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是写入还是追加
@@ -96,6 +114,21 @@ def write2nc(file, data, varname, coords, mode):
         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}  # 压缩参数
 
     # 打开 NetCDF 文件
     with nc.Dataset(file, mode, format="NETCDF4") as ncfile:
@@ -116,8 +149,17 @@ def write2nc(file, data, varname, coords, mode):
             if add_coords:
                 # 创建新坐标
                 ncfile.createDimension(dim, len(coord_data))
-                ncfile.createVariable(dim, _numpy_to_nc_type(coord_data.dtype), (dim,))
+                if compile_switch:
+                    ncfile.createVariable(dim, _numpy_to_nc_type(coord_data.dtype), (dim,), **kwargs)
+                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
@@ -127,22 +169,48 @@ def write2nc(file, data, varname, coords, mode):
                 raise ValueError("Shape of data does not match the variable shape.")
             else:
                 # 写入数据
-                ncfile.variables[varname][:] = data
+                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传入的维度名称
-            ncfile.createVariable(varname, _numpy_to_nc_type(data.dtype), dim_names)
+            if scale_offset_switch:
+                scale_factor, add_offset = _calculate_scale_and_offset(np.array(data))
+                _FillValue = -32767
+                missing_value = -32767
+                dtype = 'i2' # short类型
+            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][:] = data
+            ncfile.variables[varname][:] = 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:
+            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)
 
 
 def merge5nc(file_list, var_name=None, dim_name=None, target_filename=None):
@@ -330,6 +398,48 @@ def check_ncfile(ncfile, if_delete=False):
         return False
 
 
+def longitude_change(ds, lon_name="longitude", to_which="180"):
+    """
+    将经度转换为 -180 到 180 之间
+
+    参数：
+    lon (numpy.ndarray): 经度数组
+
+    返回值：
+    numpy.ndarray: 转换后的经度数组
+    """
+    # 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
+
+
+def nc_isel(ncfile, dim_name, slice_list):
+    """
+    Description: Choose the data by the index of the dimension
+
+    Parameters:
+        ncfile: str, the path of the netCDF file
+        dim_name: str, the name of the dimension
+        slice_list: list, the index of the dimension
+
+    slice_list example: slice_list = [[y*12+m for m in range(11,14)] for y in range(84)]
+                    or
+                        slice_list = [y * 12 + m for y in range(84) for m in range(11, 14)]
+    """
+    ds = xr.open_dataset(ncfile)
+    slice_list = np.array(slice_list).flatten()
+    slice_list = [int(i) for i in slice_list]
+    ds_new = ds.isel(**{dim_name: slice_list})
+    ds.close()
+    return ds_new
+
+
 if __name__ == "__main__":
     data = np.random.rand(100, 50)
     write2nc(r"test.nc", data, "data", {"time": np.linspace(0, 120, 100), "lev": np.linspace(0, 120, 50)}, "a")

oafuncs/oa_s/__init__.py

@@ -0,0 +1,23 @@
+#!/usr/bin/env python
+# coding=utf-8
+"""
+Author: Liu Kun && 16031215@qq.com
+Date: 2024-09-17 16:09:20
+LastEditors: Liu Kun && 16031215@qq.com
+LastEditTime: 2024-12-13 10:47:40
+FilePath: \\Python\\My_Funcs\\OAFuncs\\oafuncs\\oa_s\\__init__.py
+Description:
+EditPlatform: vscode
+ComputerInfo: XPS 15 9510
+SystemInfo: Windows 11
+Python Version: 3.11
+"""
+
+# 会导致OAFuncs直接导入所有函数，不符合模块化设计
+from .oa_cmap import *
+from .oa_data import *
+from .oa_draw import *
+from .oa_file import *
+from .oa_help import *
+from .oa_nc import *
+from .oa_python import *

oafuncs/oa_s/oa_cmap.py

@@ -0,0 +1,163 @@
+#!/usr/bin/env python
+# coding=utf-8
+"""
+Author: Liu Kun && 16031215@qq.com
+Date: 2024-09-17 16:55:11
+LastEditors: Liu Kun && 16031215@qq.com
+LastEditTime: 2024-11-21 13:14:24
+FilePath: \\Python\\My_Funcs\\OAFuncs\\oafuncs\\oa_cmap.py
+Description:
+EditPlatform: vscode
+ComputerInfo: XPS 15 9510
+SystemInfo: Windows 11
+Python Version: 3.11
+"""
+
+import matplotlib as mpl
+import matplotlib.pyplot as plt
+import numpy as np
+
+__all__ = ["show", "cmap2colors", "create_cmap", "create_cmap_rgbtxt", "choose_cmap"]
+
+# ** 将cmap用填色图可视化（官网摘抄函数）
+def show(colormaps: list):
+    """
+    Helper function to plot data with associated colormap.
+    example:
+    cmap = ListedColormap(["darkorange", "gold", "lawngreen", "lightseagreen"])
+    show([cmap])
+    """
+    np.random.seed(19680801)
+    data = np.random.randn(30, 30)
+    n = len(colormaps)
+    fig, axs = plt.subplots(1, n, figsize=(n * 2 + 2, 3), constrained_layout=True, squeeze=False)
+    for [ax, cmap] in zip(axs.flat, colormaps):
+        psm = ax.pcolormesh(data, cmap=cmap, rasterized=True, vmin=-4, vmax=4)
+        fig.colorbar(psm, ax=ax)
+    plt.show()
+
+
+# ** 将cmap转为list，即多个颜色的列表
+def cmap2colors(cmap, n=256):
+    """
+    cmap        : cmap名称
+    n           : 提取颜色数量
+    return      : 提取的颜色列表
+    example     : out_colors = cmap2colors('viridis', 256)
+    """
+    c_map = mpl.colormaps.get_cmap(cmap)
+    out_colors = [c_map(i) for i in np.linspace(0, 1, n)]
+    return out_colors
+
+
+# ** 自制cmap，多色，可带位置
+def create_cmap(colors: list, nodes=None, under=None, over=None):  # 利用颜色快速配色
+    """
+    func        : 自制cmap，自动确定颜色位置（等比例）
+    description : colors可以是颜色名称，也可以是十六进制颜色代码
+    param        {*} colors 颜色
+    param        {*} nodes 颜色位置，默认不提供，等间距
+    return       {*} cmap
+    example     : cmap = create_cmap(['#C2B7F3','#B3BBF2','#B0CBF1','#ACDCF0','#A8EEED'])
+                cmap = create_cmap(['aliceblue','skyblue','deepskyblue'],[0.0,0.5,1.0])
+    """
+    if nodes is None:  # 采取自动分配比例
+        cmap_color = mpl.colors.LinearSegmentedColormap.from_list("mycmap", colors)
+    else:  # 按照提供比例分配
+        cmap_color = mpl.colors.LinearSegmentedColormap.from_list("mycmap", list(zip(nodes, colors)))
+    if under is not None:
+        cmap_color.set_under(under)
+    if over is not None:
+        cmap_color.set_over(over)
+    return cmap_color
+
+
+# ** 根据RGB的txt文档制作色卡（利用Grads调色盘）
+def create_cmap_rgbtxt(rgbtxt_file,split_mark=','):  # 根据RGB的txt文档制作色卡/根据rgb值制作
+    """
+    func        : 根据RGB的txt文档制作色卡
+    description : rgbtxt_file='E:/python/colorbar/test.txt'
+    param        {*} rgbtxt_file txt文件路径
+    return       {*} camp
+    example     : cmap=create_cmap_rgbtxt(path,split_mark=',') #
+
+    txt example : 251,251,253
+                225,125,25
+                250,205,255
+    """
+    with open(rgbtxt_file) as fid:
+        data = fid.readlines()
+    n = len(data)
+    rgb = np.zeros((n, 3))
+    for i in np.arange(n):
+        rgb[i][0] = data[i].split(split_mark)[0]
+        rgb[i][1] = data[i].split(split_mark)[1]
+        rgb[i][2] = data[i].split(split_mark)[2]
+    max_rgb = np.max(rgb)
+    if max_rgb > 2:  # 如果rgb值大于2，则认为是0-255的值，需要归一化
+        rgb = rgb / 255.0
+    icmap = mpl.colors.ListedColormap(rgb, name="my_color")
+    return icmap
+
+
+def choose_cmap(cmap_name=None, query=False):
+    """
+    description: Choosing a colormap from the list of available colormaps or a custom colormap
+    param {*} cmap_name:
+    param {*} query:
+    return {*}
+    """
+
+    my_cmap_dict = {
+        "diverging_1": create_cmap(["#4e00b3", "#0000FF", "#00c0ff", "#a1d3ff", "#DCDCDC", "#FFD39B", "#FF8247", "#FF0000", "#FF5F9E"]),
+        "cold_1": create_cmap(["#4e00b3", "#0000FF", "#00c0ff", "#a1d3ff", "#DCDCDC"]),
+        "warm_1": create_cmap(["#DCDCDC", "#FFD39B", "#FF8247", "#FF0000", "#FF5F9E"]),
+        # "land_1": create_custom(["#3E6436", "#678A59", "#91A176", "#B8A87D", "#D9CBB2"], under="#A6CEE3", over="#FFFFFF"), # 陆地颜色从深绿到浅棕，表示从植被到沙地的递减
+        # "ocean_1": create_custom(["#126697", "#2D88B3", "#4EA1C9", "#78B9D8", "#A6CEE3"], under="#8470FF", over="#3E6436"), # 海洋颜色从深蓝到浅蓝，表示从深海到浅海的递减
+        # "ocean_land_1": create_custom(
+        #     [
+        #         "#126697",  # 深蓝（深海）
+        #         "#2D88B3",  # 蓝
+        #         "#4EA1C9",  # 蓝绿
+        #         "#78B9D8",  # 浅蓝（浅海）
+        #         "#A6CEE3",  # 浅蓝（近岸）
+        #         "#AAAAAA",  # 灰色（0值，海平面）
+        #         "#D9CBB2",  # 沙质土壤色（陆地开始）
+        #         "#B8A87D",  # 浅棕
+        #         "#91A176",  # 浅绿
+        #         "#678A59",  # 中绿
+        #         "#3E6436",  # 深绿（高山）
+        #     ]
+        # ),
+        "colorful_1": create_cmap(["#6d00db", "#9800cb", "#F2003C", "#ff4500", "#ff7f00", "#FE28A2", "#FFC0CB", "#DDA0DD", "#40E0D0", "#1a66f2", "#00f7fb", "#8fff88", "#E3FF00"]),
+    }
+    if query:
+        for key, _ in my_cmap_dict.items():
+            print(key)
+
+    if cmap_name in my_cmap_dict:
+        return my_cmap_dict[cmap_name]
+    else:
+        try:
+            return mpl.colormaps.get_cmap(cmap_name)
+        except ValueError:
+            raise ValueError(f"Unknown cmap name: {cmap_name}")
+
+
+if __name__ == "__main__":
+    # ** 测试自制cmap
+    colors = ["#C2B7F3", "#B3BBF2", "#B0CBF1", "#ACDCF0", "#A8EEED"]
+    nodes = [0.0, 0.2, 0.4, 0.6, 1.0]
+    c_map = create_cmap(colors, nodes)
+    show([c_map])
+
+    # ** 测试自制diverging型cmap
+    diverging_cmap = create_cmap(["#4e00b3", "#0000FF", "#00c0ff", "#a1d3ff", "#DCDCDC", "#FFD39B", "#FF8247", "#FF0000", "#FF5F9E"])
+    show([diverging_cmap])
+
+    # ** 测试根据RGB的txt文档制作色卡
+    file_path = "E:/python/colorbar/test.txt"
+    cmap_rgb = create_cmap_rgbtxt(file_path)
+
+    # ** 测试将cmap转为list
+    out_colors = cmap2colors("viridis", 256)

oafuncs/oa_s/oa_data.py

@@ -0,0 +1,187 @@
+#!/usr/bin/env python
+# coding=utf-8
+'''
+Author: Liu Kun && 16031215@qq.com
+Date: 2024-09-17 17:12:47
+LastEditors: Liu Kun && 16031215@qq.com
+LastEditTime: 2024-11-21 13:13:20
+FilePath: \\Python\\My_Funcs\\OAFuncs\\oafuncs\\oa_data.py
+Description:  
+EditPlatform: vscode
+ComputerInfo: XPS 15 9510
+SystemInfo: Windows 11
+Python Version: 3.11
+'''
+
+
+import multiprocessing as mp
+from concurrent.futures import ThreadPoolExecutor
+
+import numpy as np
+from scipy.interpolate import griddata
+
+__all__ = ['interp_2d', 'interp_2d_parallel']
+
+# ** 高维插值函数，插值最后两个维度
+
+
+def interp_2d(target_x, target_y, origin_x, origin_y, data, method='linear'):
+    """
+    高维插值函数，默认插值最后两个维度，传输数据前请确保数据的维度正确
+    参数:
+    target_y (array-like): 目标经度网格 1D 或 2D
+    target_x (array-like): 目标纬度网格 1D 或 2D
+    origin_y (array-like): 初始经度网格 1D 或 2D
+    origin_x (array-like): 初始纬度网格 1D 或 2D
+    data (array-like): 数据 (*, lat, lon) 2D, 3D, 4D
+    method (str, optional): 插值方法，可选 'linear', 'nearest', 'cubic' 等，默认为 'linear'
+    返回:
+    array-like: 插值结果
+    """
+
+    # 确保目标网格和初始网格都是二维的
+    if len(target_y.shape) == 1:
+        target_x, target_y = np.meshgrid(target_x, target_y)
+    if len(origin_y.shape) == 1:
+        origin_x, origin_y = np.meshgrid(origin_x, origin_y)
+
+    dims = data.shape
+    len_dims = len(dims)
+    # print(dims[-2:])
+    # 根据经纬度网格判断输入数据的形状是否匹配
+
+    if origin_x.shape != dims[-2:] or origin_y.shape != dims[-2:]:
+        print(origin_x.shape, dims[-2:])
+        raise ValueError('Shape of data does not match shape of origin_x or origin_y.')
+
+    # 将目标网格展平成一维数组
+    target_points = np.column_stack((np.ravel(target_y), np.ravel(target_x)))
+
+    # 将初始网格展平成一维数组
+    origin_points = np.column_stack((np.ravel(origin_y), np.ravel(origin_x)))
+
+    # 进行插值
+    if len_dims == 2:
+        interpolated_data = griddata(origin_points, np.ravel(data), target_points, method=method)
+        interpolated_data = np.reshape(interpolated_data, target_y.shape)
+    elif len_dims == 3:
+        interpolated_data = []
+        for i in range(dims[0]):
+            dt = griddata(origin_points, np.ravel(data[i, :, :]), target_points, method=method)
+            interpolated_data.append(np.reshape(dt, target_y.shape))
+            print(f'Interpolating {i+1}/{dims[0]}...')
+        interpolated_data = np.array(interpolated_data)
+    elif len_dims == 4:
+        interpolated_data = []
+        for i in range(dims[0]):
+            interpolated_data.append([])
+            for j in range(dims[1]):
+                dt = griddata(origin_points, np.ravel(data[i, j, :, :]), target_points, method=method)
+                interpolated_data[i].append(np.reshape(dt, target_y.shape))
+                print(f'\rInterpolating {i*dims[1]+j+1}/{dims[0]*dims[1]}...', end='')
+        print('\n')
+        interpolated_data = np.array(interpolated_data)
+
+    return interpolated_data
+
+
+# ** 高维插值函数，插值最后两个维度，使用多线程进行插值
+# 在本地电脑上可以提速三倍左右，超算上暂时无法加速
+def interp_2d_parallel(target_x, target_y, origin_x, origin_y, data, method='linear'):
+    '''
+    param        {*} target_x 目标经度网格 1D 或 2D
+    param        {*} target_y 目标纬度网格 1D 或 2D
+    param        {*} origin_x 初始经度网格 1D 或 2D
+    param        {*} origin_y 初始纬度网格 1D 或 2D
+    param        {*} data 数据 (*, lat, lon) 2D, 3D, 4D
+    param        {*} method 插值方法，可选 'linear', 'nearest', 'cubic' 等，默认为 'linear'
+    return       {*} 插值结果
+    description : 高维插值函数，默认插值最后两个维度，传输数据前请确保数据的维度正确
+    example     : interpolated_data = interp_2d_parallel(target_x, target_y, origin_x, origin_y, data, method='linear')
+    '''
+    def interp_single2d(target_y, target_x, origin_y, origin_x, data, method='linear'):
+        target_points = np.column_stack((np.ravel(target_y), np.ravel(target_x)))
+        origin_points = np.column_stack((np.ravel(origin_y), np.ravel(origin_x)))
+
+        dt = griddata(origin_points, np.ravel(data[:, :]), target_points, method=method)
+        return np.reshape(dt, target_y.shape)
+
+    def interp_single3d(i, target_y, target_x, origin_y, origin_x, data, method='linear'):
+        target_points = np.column_stack((np.ravel(target_y), np.ravel(target_x)))
+        origin_points = np.column_stack((np.ravel(origin_y), np.ravel(origin_x)))
+
+        dt = griddata(origin_points, np.ravel(data[i, :, :]), target_points, method=method)
+        return np.reshape(dt, target_y.shape)
+
+    def interp_single4d(i, j, target_y, target_x, origin_y, origin_x, data, method='linear'):
+        target_points = np.column_stack((np.ravel(target_y), np.ravel(target_x)))
+        origin_points = np.column_stack((np.ravel(origin_y), np.ravel(origin_x)))
+
+        dt = griddata(origin_points, np.ravel(data[i, j, :, :]), target_points, method=method)
+        return np.reshape(dt, target_y.shape)
+
+    if len(target_y.shape) == 1:
+        target_x, target_y = np.meshgrid(target_x, target_y)
+    if len(origin_y.shape) == 1:
+        origin_x, origin_y = np.meshgrid(origin_x, origin_y)
+
+    dims = data.shape
+    len_dims = len(dims)
+
+    if origin_x.shape != dims[-2:] or origin_y.shape != dims[-2:]:
+        raise ValueError('数据形状与 origin_x 或 origin_y 的形状不匹配.')
+
+    interpolated_data = []
+
+    # 使用多线程进行插值
+    with ThreadPoolExecutor(max_workers=mp.cpu_count()-2) as executor:
+        print(f'Using {mp.cpu_count()-2} threads...')
+        if len_dims == 2:
+            interpolated_data = list(executor.map(interp_single2d, [target_y], [target_x], [origin_y], [origin_x], [data], [method]))
+        elif len_dims == 3:
+            interpolated_data = list(executor.map(interp_single3d, [i for i in range(dims[0])], [target_y]*dims[0], [target_x]*dims[0], [origin_y]*dims[0], [origin_x]*dims[0], [data]*dims[0], [method]*dims[0]))
+        elif len_dims == 4:
+            interpolated_data = list(executor.map(interp_single4d, [i for i in range(dims[0]) for j in range(dims[1])], [j for i in range(dims[0]) for j in range(dims[1])], [target_y]*dims[0]*dims[1], [target_x]*dims[0]*dims[1], [origin_y]*dims[0]*dims[1], [origin_x]*dims[0]*dims[1], [data]*dims[0]*dims[1], [method]*dims[0]*dims[1]))
+            interpolated_data = np.array(interpolated_data).reshape(dims[0], dims[1], target_y.shape[0], target_x.shape[1])
+
+    interpolated_data = np.array(interpolated_data)
+
+    return interpolated_data
+
+
+if __name__ == '__main__':
+    import time
+
+    import matplotlib.pyplot as plt
+
+    # 测试数据
+    origin_x = np.linspace(0, 10, 11)
+    origin_y = np.linspace(0, 10, 11)
+    target_x = np.linspace(0, 10, 101)
+    target_y = np.linspace(0, 10, 101)
+    data = np.random.rand(11, 11)
+
+    # 高维插值
+    origin_x = np.linspace(0, 10, 11)
+    origin_y = np.linspace(0, 10, 11)
+    target_x = np.linspace(0, 10, 101)
+    target_y = np.linspace(0, 10, 101)
+    data = np.random.rand(10, 10, 11, 11)
+
+    start = time.time()
+    interpolated_data = interp_2d(target_x, target_y, origin_x, origin_y, data)
+    print(f'Interpolation time: {time.time()-start:.2f}s')
+
+    print(interpolated_data.shape)
+
+    # 高维插值多线程
+    start = time.time()
+    interpolated_data = interp_2d_parallel(target_x, target_y, origin_x, origin_y, data)
+    print(f'Interpolation time: {time.time()-start:.2f}s')
+
+    print(interpolated_data.shape)
+    print(interpolated_data[0, 0, :, :].shape)
+    plt.figure()
+    plt.contourf(target_x, target_y, interpolated_data[0, 0, :, :])
+    plt.colorbar()
+    plt.show()