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

oafuncs/oa_draw.py

@@ -1,18 +1,17 @@
 #!/usr/bin/env python
 # coding=utf-8
-'''
+"""
 Author: Liu Kun && 16031215@qq.com
 Date: 2024-09-17 17:26:11
 LastEditors: Liu Kun && 16031215@qq.com
 LastEditTime: 2024-11-21 13:10:47
 FilePath: \\Python\\My_Funcs\\OAFuncs\\oafuncs\\oa_draw.py
-Description:  
+Description:
 EditPlatform: vscode
 ComputerInfo: XPS 15 9510
 SystemInfo: Windows 11
 Python Version: 3.11
-'''
-
+"""
 
 import math
 import warnings
@@ -26,22 +25,23 @@ import xarray as xr
 from cartopy.mpl.ticker import LatitudeFormatter, LongitudeFormatter
 from mpl_toolkits.axes_grid1 import make_axes_locatable
 
-__all__ = ['fig_minus', 'create_gif', 'xy2lonlat', 'plot_contourf', 'plot_contourf_lonlat', 'plot_quiver', 'plot_contourf_cartopy']
+__all__ = ["fig_minus", "create_gif", "add_cartopy", "add_gridlines", 'MidpointNormalize', "xy2lonlat", "plot_contourf", "plot_contourf_lonlat", "plot_quiver", "plot_contourf_cartopy"]
+
+warnings.filterwarnings("ignore")
 
-warnings.filterwarnings('ignore')
 
 def fig_minus(ax_x=None, ax_y=None, cbar=None, decimal=None, add_space=False):
-    '''
+    """
     Description: 将坐标轴刻度中的负号替换为减号
-    
+
     param {*} ax_x : x轴
     param {*} ax_y : y轴
     param {*} cbar : colorbar
     param {*} decimal : 小数位数
     param {*} add_space : 是否在非负数前面加空格
-    
+
     return {*} ax_x or ax_y or cbar
-    '''
+    """
     if ax_x is not None:
         current_ticks = ax_x.get_xticks()
     if ax_y is not None:
@@ -74,37 +74,38 @@ def fig_minus(ax_x=None, ax_y=None, cbar=None, decimal=None, add_space=False):
         cbar.set_ticklabels(out_ticks)
         return cbar
 
-# ** 将生成图片/已有图片制作成动图
-
 
+# ** 将生成图片/已有图片制作成动图
 def create_gif(image_list: list, gif_name: str, duration=0.2):  # 制作动图，默认间隔0.2
-    '''
+    """
     func        : 制作动图，将已有图片拼接
     description : Gif格式动图
     param        {*} image_list 图片列表
     param        {*} gif_name 动图名称（含路径）
     param        {*} duration 动图间隔
     return       {*} 自动保存至指定路径（包含于动图名称中）
-    example     : 
-    '''
+    example     :
+    """
     import imageio.v2 as imageio
+
     frames = []
     for image_name in image_list:
         frames.append(imageio.imread(image_name))
-    imageio.mimsave(gif_name, frames, format='GIF', duration=duration)
-    print('Gif制作完成！')
+    imageio.mimsave(gif_name, frames, format="GIF", duration=duration)
+    print("Gif制作完成！")
     return
 
 
 # ** 转化经/纬度刻度
-def xy2lonlat(xy, lonlat='lon', decimal=2):
-    '''
+def xy2lonlat(xy, lonlat="lon", decimal=2):
+    """
     param        {*} xy : 经/纬度列表
     param        {*} lonlat : 'lon' or 'lat'
     param        {*} decimal : 小数位数
     return       {*} 转化后的经/纬度列表
     example     : xy2lonlat(x, lonlat='lon', decimal=2)
-    '''
+    """
+
     def format_longitude(x_list):
         out_list = []
         for x in x_list:
@@ -127,9 +128,9 @@ def xy2lonlat(xy, lonlat='lon', decimal=2):
             out_list.append(f"{degrees:.{decimal}f}°{direction}" if y != 0 else f"{degrees}°")
         return out_list if len(out_list) > 1 else out_list[0]
 
-    if lonlat == 'lon':
+    if lonlat == "lon":
         return format_longitude(xy)
-    elif lonlat == 'lat':
+    elif lonlat == "lat":
         return format_latitude(xy)
 
 
@@ -142,32 +143,31 @@ class _MyFormatter(mpl.ticker.ScalarFormatter):
         self.vmin = min(cticks)
         self.vmax = max(cticks)
         self.p_n = self.vmin < 0 and self.vmax > 0
-        self.magnitude_min = int(
-            math.modf(math.log10(min(abs(cticks[cticks != 0]))))[1])
+        self.magnitude_min = int(math.modf(math.log10(min(abs(cticks[cticks != 0]))))[1])
         self.magnitude_max = int(math.modf(math.log10(max(abs(cticks))))[1])
         # print(self.vmin, self.vmax)
 
     def __call__(self, x, pos):
         if ((abs(x) < 1e-2) or (abs(x) > 1e4)) and x != 0:
             if self.magnitude_max - self.magnitude_min == 1 and (int(math.modf(math.log10(abs(x)))[1]) == self.magnitude_min):
-                a, b = '{:.1e}'.format(x).split('e')
+                a, b = "{:.1e}".format(x).split("e")
                 a = float(a) / 10
                 b = int(b) + 1
             else:
-                a, b = '{:.2e}'.format(x).split('e')
+                a, b = "{:.2e}".format(x).split("e")
                 a = float(a)
                 b = int(b)
             # return '${}{} \\times 10^{{{}}}$'.format(' ' if (self.p_n and x > 0) else '', a, b)
-            return '${}{:.2f}$'.format(' ' if (self.p_n and x > 0) else '', a)
+            return "${}{:.2f}$".format(" " if (self.p_n and x > 0) else "", a)
         elif x == 0:
-            return '0'
+            return "0"
         else:
             return mpl.ticker.ScalarFormatter.__call__(self, x, pos)
 
 
 # ** 绘制单张填色图
-def plot_contourf(pic_data, picname=None, c_map='rainbow', minmax=None, labels=None, ticks_space=None, ticks=None, figsize=(12, 9)):
-    '''
+def plot_contourf(pic_data, picname=None, c_map="rainbow", minmax=None, labels=None, ticks_space=None, ticks=None, figsize=(12, 9)):
+    """
     func        : 绘制填色等值线图，单张
     description : 绘制单张填色等值线图，输入参数为横纵坐标、等值数据、图例标注等
     param       {*}pic_data : 填色等值线图的等值数据
@@ -178,7 +178,7 @@ def plot_contourf(pic_data, picname=None, c_map='rainbow', minmax=None, labels=N
     param       {*}ticks_space : x、y轴刻度，以及所显示的标签，默认不显示
     param       {*}figsize  : 图片大小，默认(12,9)
     example     : plot_contourf(pic_data, pictpath, var_name, c_map='bwr', labels=None, ticks_space=None, ticks=None, h=0, figsize=(12, 9))
-    '''
+    """
     cmap = mpl.colormaps.get_cmap(c_map)
     if minmax is not None:
         value_min, value_max = minmax[0], minmax[1]
@@ -186,8 +186,7 @@ def plot_contourf(pic_data, picname=None, c_map='rainbow', minmax=None, labels=N
         value_min, value_max = pic_data.nanmin(), pic_data.nanmax()
     v_bry = max(abs(value_min), abs(value_max))
     flag = (value_min < 0) and (value_max > 0)
-    norm = mpl.colors.TwoSlopeNorm(
-        vmin=-1 * v_bry, vcenter=0, vmax=v_bry) if flag else mpl.colors.Normalize(vmin=value_min, vmax=value_max)
+    norm = mpl.colors.TwoSlopeNorm(vmin=-1 * v_bry, vcenter=0, vmax=v_bry) if flag else mpl.colors.Normalize(vmin=value_min, vmax=value_max)
     cticks = [num for num in np.linspace(-1 * v_bry if flag else value_min, v_bry if flag else value_max, 9)] if value_min != value_max else None
     levels = np.linspace(-1 * v_bry, v_bry, 20) if flag else None if value_min == value_max else np.linspace(value_min, value_max, 20)
 
@@ -196,8 +195,8 @@ def plot_contourf(pic_data, picname=None, c_map='rainbow', minmax=None, labels=N
 
     fig, ax = plt.subplots(figsize=figsize)
     flag_lc = levels is not None and cticks is not None
-    CS = ax.contourf(x, y, pic_data, cmap=cmap, norm=norm, levels=levels, extend='both') if flag_lc else ax.contourf(x, y, pic_data, cmap=cmap, norm=norm, extend='both')
-    cb = fig.colorbar(CS, ax=ax, orientation='vertical', shrink=1, format='%.3g', spacing='uniform', ticks=cticks) if cticks is not None else fig.colorbar(CS, ax=ax, orientation='vertical', shrink=1, format='%.3g', spacing='uniform')
+    CS = ax.contourf(x, y, pic_data, cmap=cmap, norm=norm, levels=levels, extend="both") if flag_lc else ax.contourf(x, y, pic_data, cmap=cmap, norm=norm, extend="both")
+    cb = fig.colorbar(CS, ax=ax, orientation="vertical", shrink=1, format="%.3g", spacing="uniform", ticks=cticks) if cticks is not None else fig.colorbar(CS, ax=ax, orientation="vertical", shrink=1, format="%.3g", spacing="uniform")
     """%.3g采用的是自动调整格式，也可设置为%.3f，则改为3位小数"""
 
     # 将格式化器设置为自定义的函数
@@ -209,20 +208,16 @@ def plot_contourf(pic_data, picname=None, c_map='rainbow', minmax=None, labels=N
         cb.update_ticks()
 
     if labels is not None:
-        cb.set_label(labels['c'])
-        plt.xlabel(labels['x'])
-        plt.ylabel(labels['y'])
+        cb.set_label(labels["c"])
+        plt.xlabel(labels["x"])
+        plt.ylabel(labels["y"])
     if ticks_space is not None and ticks is not None:
-        plt.xticks(np.arange(0, len(x[0, :]) +
-                   1e-5, ticks_space['x']), ticks['x'])
-        plt.yticks(np.arange(0, len(y[:, 0]) +
-                   1e-5, ticks_space['y']), ticks['y'])
+        plt.xticks(np.arange(0, len(x[0, :]) + 1e-5, ticks_space["x"]), ticks["x"])
+        plt.yticks(np.arange(0, len(y[:, 0]) + 1e-5, ticks_space["y"]), ticks["y"])
 
-    plt.title('Min: {:.3g}, Max: {:.3g}'.format(
-        pic_data.min(), pic_data.max()))
+    plt.title("Min: {:.3g}, Max: {:.3g}".format(pic_data.min(), pic_data.max()))
 
-    plt.savefig(
-        picname, bbox_inches='tight') if picname is not None else plt.show()
+    plt.savefig(picname, bbox_inches="tight") if picname is not None else plt.show()
     # plt.show()
     plt.clf()
     # 关闭当前figure
@@ -230,8 +225,8 @@ def plot_contourf(pic_data, picname=None, c_map='rainbow', minmax=None, labels=N
 
 
 # ** 画等高线图，带经纬度坐标轴
-def plot_contourf_lonlat(data, lon, lat, interval=5, picname=None, c_map='rainbow'):
-    '''
+def plot_contourf_lonlat(data, lon, lat, interval=5, picname=None, c_map="rainbow"):
+    """
     param        {*} data : 二维数据
     param        {*} lon : 经度
     param        {*} lat : 纬度
@@ -239,7 +234,7 @@ def plot_contourf_lonlat(data, lon, lat, interval=5, picname=None, c_map='rainbo
     param        {*} picname : 图片保存的文件名(含路径)
     param        {*} c_map : 颜色映射，默认rainbow
     return       {*} 无返回值
-    '''
+    """
     if len(lon.shape) == 2:
         lon = lon[0, :]
     if len(lat.shape) == 2:
@@ -267,14 +262,13 @@ def plot_contourf_lonlat(data, lon, lat, interval=5, picname=None, c_map='rainbo
     plt.xticks(np.arange(0, len(lon), x_space), [format_longitude(lon[i]) for i in range(0, len(lon), x_space)])
     plt.yticks(np.arange(0, len(lat), y_space), [format_latitude(lat[i]) for i in range(0, len(lat), y_space)])
     plt.colorbar()
-    plt.savefig(
-        picname, bbox_inches='tight') if picname is not None else plt.show()
+    plt.savefig(picname, bbox_inches="tight") if picname is not None else plt.show()
     plt.close()
 
 
 # ** 绘制矢量场
-def plot_quiver(u, v, lon, lat, picname=None, cmap='coolwarm', scale=0.25, width=0.002, x_space=5, y_space=5):
-    '''
+def plot_quiver(u, v, lon, lat, picname=None, cmap="coolwarm", scale=0.25, width=0.002, x_space=5, y_space=5):
+    """
     param        {*} u : 二维数据
     param        {*} v : 二维数据
     param        {*} lon : 经度, 1D or 2D
@@ -286,7 +280,7 @@ def plot_quiver(u, v, lon, lat, picname=None, cmap='coolwarm', scale=0.25, width
     param        {*} x_space : x轴间隔
     param        {*} y_space : y轴间隔
     return       {*} 无返回值
-    '''
+    """
     # 创建新的网格位置变量(lat_c, lon_c)
     if len(lon.shape) == 1 and len(lat.shape) == 1:
         lon_c, lat_c = np.meshgrid(lon, lat)
@@ -307,29 +301,31 @@ def plot_quiver(u, v, lon, lat, picname=None, cmap='coolwarm', scale=0.25, width
 
     # 使用 plt.quiver 函数绘制矢量图
     # 通过设置 quiver 函数的 pivot 参数来指定箭头的位置
-    quiver_plot = plt.quiver(lon_c[::y_space, ::x_space],
-                             lat_c[::y_space, ::x_space],
-                             u[::y_space, ::x_space],
-                             v[::y_space, ::x_space],
-                             S[::y_space, ::x_space],  # 矢量的大小，可以不要
-                             pivot='middle',
-                             scale=scale,
-                             #  color=color, # 矢量的颜色，单色
-                             cmap=cmap,  # 矢量的颜色，多色
-                             width=width)
+    quiver_plot = plt.quiver(
+        lon_c[::y_space, ::x_space],
+        lat_c[::y_space, ::x_space],
+        u[::y_space, ::x_space],
+        v[::y_space, ::x_space],
+        S[::y_space, ::x_space],  # 矢量的大小，可以不要
+        pivot="middle",
+        scale=scale,
+        #  color=color, # 矢量的颜色，单色
+        cmap=cmap,  # 矢量的颜色，多色
+        width=width,
+    )
     # plt.quiverkey(quiver_plot, X=0.90, Y=0.975, U=1, label='1 m/s', labelpos='E', fontproperties={'size': 10})
-    plt.quiverkey(quiver_plot, X=0.87, Y=0.975, U=mean_S, label=f'{mean_S:.2f} m/s', labelpos='E', fontproperties={'size': 10})
+    plt.quiverkey(quiver_plot, X=0.87, Y=0.975, U=mean_S, label=f"{mean_S:.2f} m/s", labelpos="E", fontproperties={"size": 10})
     plt.colorbar(quiver_plot)
-    plt.xlabel('X')
-    plt.ylabel('Y')
+    plt.xlabel("X")
+    plt.ylabel("Y")
 
-    plt.savefig(picname, bbox_inches='tight') if picname is not None else plt.show()
+    plt.savefig(picname, bbox_inches="tight") if picname is not None else plt.show()
     plt.clf()
     plt.close()
 
 
-def plot_contourf_cartopy(data, lon, lat, picname=None, cmap='rainbow', cn_fill_num=20, fig_size=(12, 9), title='Cartopy', land_color='green', ocean_color='lightgrey'):
-    '''
+def plot_contourf_cartopy(data, lon, lat, picname=None, cmap="rainbow", cn_fill_num=20, fig_size=(12, 9), title="Cartopy", land_color="green", ocean_color="lightgrey"):
+    """
     param        {*} data : 二维数据
     param        {*} lon : 经度
     param        {*} lat : 纬度
@@ -341,7 +337,7 @@ def plot_contourf_cartopy(data, lon, lat, picname=None, cmap='rainbow', cn_fill_
     param        {*} land_color : 陆地颜色
     param        {*} ocean_color : 海洋颜色
     return       {*} 无返回值
-    '''
+    """
     if len(lon.shape) == 2:
         lon = lon[0, :]
     if len(lat.shape) == 2:
@@ -381,37 +377,33 @@ def plot_contourf_cartopy(data, lon, lat, picname=None, cmap='rainbow', cn_fill_
     cticks = cbar_ticks
     norm = mpl.colors.BoundaryNorm(cticks, cmap.N)
 
-    cnplot = ax.contourf(X, Y, data, levels=levels, cmap=cmap, norm=norm, transform=proj, extend='both', alpha=1, zorder=0)
+    cnplot = ax.contourf(X, Y, data, levels=levels, cmap=cmap, norm=norm, transform=proj, extend="both", alpha=1, zorder=0)
     # cllevels = np.linspace(data_min, data_max, 9)
     # clplot = ax.contour(X, Y, data, levels=levels[9::10], colors='k', linewidths=0.5, transform=proj, zorder=1, alpha=0.8, linestyle='--')
     # 添加色标，并选择位置
     divider = make_axes_locatable(ax)
     location = 3
     if location == 1:  # 左侧
-        cax = divider.new_horizontal(
-            size="5%", pad=1, axes_class=plt.Axes, pack_start=True)
+        cax = divider.new_horizontal(size="5%", pad=1, axes_class=plt.Axes, pack_start=True)
         fig.add_axes(cax)
-        cbar = plt.colorbar(
-            cnplot, cax=cax, orientation='vertical', extend='both')
+        cbar = plt.colorbar(cnplot, cax=cax, orientation="vertical", extend="both")
     elif location == 2:  # 下方
-        cax = divider.new_vertical(
-            size="5%", pad=0.3, axes_class=plt.Axes, pack_start=True)
+        cax = divider.new_vertical(size="5%", pad=0.3, axes_class=plt.Axes, pack_start=True)
         fig.add_axes(cax)
-        cbar = plt.colorbar(
-            cnplot, cax=cax, orientation='horizontal', extend='both')
+        cbar = plt.colorbar(cnplot, cax=cax, orientation="horizontal", extend="both")
     elif location == 3:  # 右侧
         cax = divider.new_horizontal(size="5%", pad=0.1, axes_class=plt.Axes)
         fig.add_axes(cax)
         # cbar = plt.colorbar(cnplot, cax=cax, orientation='vertical', extend='both', format='%.0f')
-        cbar = fig.colorbar(mpl.cm.ScalarMappable(cmap=cmap, norm=norm), cax=cax, orientation='vertical', extend='both', format='%.3f')
-        cax.yaxis.set_ticks_position('right')
-        cax.yaxis.set_label_position('right')
+        cbar = fig.colorbar(mpl.cm.ScalarMappable(cmap=cmap, norm=norm), cax=cax, orientation="vertical", extend="both", format="%.3f")
+        cax.yaxis.set_ticks_position("right")
+        cax.yaxis.set_label_position("right")
     else:  # 上方
         cax = divider.new_vertical(size="5%", pad=0.2, axes_class=plt.Axes)
         fig.add_axes(cax)
-        cbar = plt.colorbar(cnplot, cax=cax, orientation='horizontal', extend='both')
+        cbar = plt.colorbar(cnplot, cax=cax, orientation="horizontal", extend="both")
     cbar.ax.tick_params(labelsize=10)
-    cbar.ax.xaxis.set_tick_params(direction='in', width=1, length=2)
+    cbar.ax.xaxis.set_tick_params(direction="in", width=1, length=2)
     # 添加cbar_ticks
     # cbar.set_ticks(np.arange(round(levels[0]), round(levels[-1]), round((levels[-1]-levels[0])/9)))  # 设置色标刻度
     # cbar.set_ticks(cbar_ticks)  # 设置色标刻度
@@ -421,31 +413,91 @@ def plot_contourf_cartopy(data, lon, lat, picname=None, cmap='rainbow', cn_fill_
     # cbar.set_ticks(np.arange(round(levels[0]), round(levels[-1]), round((levels[-1]-levels[0])/9)))  # 设置色标刻度
 
     # 单独设置label
-    cbar.set_label(title, fontsize=10, weight='bold')
+    cbar.set_label(title, fontsize=10, weight="bold")
     # cax.set_position([0.1, 0.2, 0.02, 0.6]) # 调整色标位置
-    fig.savefig(picname, bbox_inches='tight', dpi=600) if picname is not None else plt.show()
+    fig.savefig(picname, bbox_inches="tight", dpi=600) if picname is not None else plt.show()
     plt.close()
 
 
-if __name__ == '__main__':
+def add_gridlines(ax, projection=ccrs.PlateCarree(), color="k", alpha=0.5, linestyle="--", linewidth=0.5):
+    # add gridlines
+    gl = ax.gridlines(crs=projection, draw_labels=True, linewidth=linewidth, color=color, alpha=alpha, linestyle=linestyle)
+    gl.right_labels = False
+    gl.top_labels = False
+    gl.xformatter = LongitudeFormatter(zero_direction_label=False)
+    gl.yformatter = LatitudeFormatter()
+    
+    return ax, gl
+
+
+def add_cartopy(ax,lon=None,lat=None,projection=ccrs.PlateCarree(), gridlines=True,landcolor="lightgrey",oceancolor="lightblue", cartopy_linewidth=0.5):
+    # Set the projection for the axes
+    ax.set_projection(projection)
+    
+    # add coastlines
+    ax.add_feature(cfeature.LAND, facecolor=landcolor)
+    ax.add_feature(cfeature.OCEAN, facecolor=oceancolor)
+    ax.add_feature(cfeature.COASTLINE, linewidth=cartopy_linewidth)
+    ax.add_feature(cfeature.BORDERS, linewidth=cartopy_linewidth, linestyle=":")
+
+    # add gridlines
+    if gridlines:
+        ax, gl = add_gridlines(ax, projection)
+    
+    # set longitude and latitude format
+    lon_formatter = LongitudeFormatter(zero_direction_label=False)
+    lat_formatter = LatitudeFormatter()
+    ax.xaxis.set_major_formatter(lon_formatter)
+    ax.yaxis.set_major_formatter(lat_formatter)
+    
+    # set extent
+    if lon is not None and lat is not None:
+        lon_min, lon_max = lon.min(), lon.max()
+        lat_min, lat_max = lat.min(), lat.max()
+        ax.set_extent([lon_min, lon_max, lat_min, lat_max], crs=projection)
+
+
+class MidpointNormalize(mpl.colors.Normalize):
+    '''
+    Description: 自定义归一化类，使得0值处为中心点
+    
+    param {*} mpl.colors.Normalize : 继承Normalize类
+    return {*}
+    
+    Example:
+    nrom = MidpointNormalize(vmin=-2, vmax=1, vcenter=0)
+    '''
+    def __init__(self, vmin=None, vmax=None, vcenter=None, clip=False):
+        self.vcenter = vcenter
+        super().__init__(vmin, vmax, clip)
+
+    def __call__(self, value, clip=None):
+        x, y = [self.vmin, self.vcenter, self.vmax], [0, 0.5, 1.]
+        return np.ma.masked_array(np.interp(value, x, y, left=-np.inf, right=np.inf))
+
+    def inverse(self, value):
+        y, x = [self.vmin, self.vcenter, self.vmax], [0, 0.5, 1]
+        return np.interp(value, x, y, left=-np.inf, right=np.inf)
+
+if __name__ == "__main__":
     # ** 绘制填色图
     data = np.random.randn(100, 100)
-    picname = 'test.png'
-    plot_contourf(data, picname, c_map='rainbow', minmax=None, labels=None, ticks_space=None, ticks=None, figsize=(12, 9))
+    picname = "test.png"
+    plot_contourf(data, picname, c_map="rainbow", minmax=None, labels=None, ticks_space=None, ticks=None, figsize=(12, 9))
     # ** 绘制矢量场
     u = np.random.randn(100, 100)
     v = np.random.randn(100, 100)
     lon = np.linspace(0, 360, 100)
     lat = np.linspace(-90, 90, 100)
-    picname = 'test.png'
-    plot_quiver(u, v, lon, lat, picname, cmap='coolwarm', scale=0.25, width=0.002, x_space=5, y_space=5)
+    picname = "test.png"
+    plot_quiver(u, v, lon, lat, picname, cmap="coolwarm", scale=0.25, width=0.002, x_space=5, y_space=5)
     # ** 绘制经纬度填色图
     data = np.random.randn(100, 100)
     lon = np.linspace(0, 360, 100)
     lat = np.linspace(-90, 90, 100)
-    picname = 'test.png'
-    plot_contourf_lonlat(data, lon, lat, interval=5, picname=picname, c_map='rainbow')
+    picname = "test.png"
+    plot_contourf_lonlat(data, lon, lat, interval=5, picname=picname, c_map="rainbow")
     # ** 制作动图
-    image_list = ['test1.png', 'test2.png', 'test3.png']
-    gif_name = 'test.gif'
+    image_list = ["test1.png", "test2.png", "test3.png"]
+    gif_name = "test.gif"
     create_gif(image_list, gif_name, duration=0.2)