shancx 1.9.33.109__py3-none-any.whl → 1.9.33.218__py3-none-any.whl

shancx/Plot/__init__.py

@@ -21,9 +21,9 @@ def plotGrey(img,name="plotGrey", saveDir="plotGrey",cmap='gray', title='Image')
 import matplotlib.pyplot as plt
 from shancx import crDir
 import datetime
-def plotMat(matrix,name='plotMat',saveDir="plotMat",title='Matrix Plot', xlabel='X-axis', ylabel='Y-axis', color_label='Value', cmap='viridis'):
+def plotMat(matrix,name='plotMat',saveDir="plotMat",title='Matrix Plot', xlabel='X-axis', ylabel='Y-axis', color_label='Value', cmap='viridis',aspect="equal"): #aspect='auto'
     now_str = datetime.datetime.now().strftime("%Y%m%d%H%M%S")    
-    plt.imshow(matrix, cmap=cmap, origin='upper', aspect='auto')
+    plt.imshow(matrix, cmap=cmap, origin='upper', aspect=f'{aspect}')
     plt.colorbar(label=color_label)
     plt.title(title)
     plt.xlabel(xlabel)
@@ -34,6 +34,35 @@ def plotMat(matrix,name='plotMat',saveDir="plotMat",title='Matrix Plot', xlabel=
     plt.savefig(outpath)
     plt.close()
 
+import matplotlib.pyplot as plt
+from shancx import crDir
+import datetime
+def plotMatplus(matrix, name='plotMat', saveDir="plotMat", title='Matrix Plot', 
+           xlabel='Longitude', ylabel='Latitude', color_label='Value', 
+           cmap='viridis', extent=None):
+    """
+    extent: [lon_min, lon_max, lat_min, lat_max]
+    """
+    now_str = datetime.datetime.now().strftime("%Y%m%d%H%M%S")    
+    plt.imshow(matrix, cmap=cmap, origin='upper', aspect='auto', extent=extent)
+    plt.colorbar(label=color_label)
+    plt.title(title)
+    
+    # 添加度符号和方向标识
+    plt.xlabel(f'{xlabel} (°E)')  # 东经
+    plt.ylabel(f'{ylabel} (°N)')  # 北纬
+    
+    plt.tight_layout()
+    outpath = f'./{saveDir}/{name}_{now_str}.png' if name=="plotMat" else f"./{saveDir}/{name}.png"
+    crDir(outpath)
+    plt.savefig(outpath)
+    plt.close()
+"""
+latlon = [10.0, 37.0, 105.0, 125.0] 
+latmin, latmax, lonmin, lonmax = latlon 
+plotMatplus(data,extent=[lon_min, lon_max, lat_min, lat_max]) 
+"""
+
 import datetime
 from hjnwtx.colormap import cmp_hjnwtx
 from shancx import crDir
@@ -45,7 +74,6 @@ def plotRadar(array_dt,name="plotRadar", saveDir="plotRadar",ty="CR"):
     if len(array_dt.shape) == 2 and ty == "pre":
         fig, ax = plt.subplots()
         im = ax.imshow(array_dt, vmin=0, vmax=10, cmap=cmp_hjnwtx["pre_tqw"])        
-        # 创建与图像高度一致的colorbar
         divider = make_axes_locatable(ax)
         cax = divider.append_axes("right", size="5%", pad=0.05)
         plt.colorbar(im, cax=cax)   
@@ -122,6 +150,27 @@ def plotScatter(df1,saveDir="plotScatter"):
     plt.savefig(f"./{saveDir}/plotScatter_{now_str}.png", dpi=300, bbox_inches="tight")  
     plt.close()
 
+import matplotlib.pyplot as plt
+import os
+def plotScatter1(true,pre,saveDir="plotScatter"):
+    now_str = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
+    plt.figure(figsize=(10, 8))   
+    plt.scatter(
+        true,  
+        pre,  
+        s=25,        
+        alpha=0.6,   
+        edgecolor="black",   
+        linewidth=0.5       
+    )
+    plt.title("Scatter Plot of Ture Pre", fontsize=14)
+    plt.xlabel("Longitude", fontsize=12)
+    plt.ylabel("Latitude", fontsize=12)
+    plt.tight_layout()  
+    os.makedirs(saveDir, exist_ok=True)
+    plt.savefig(f"./{saveDir}/plotScatter1_{now_str}.png", dpi=300, bbox_inches="tight")  
+    plt.close()
+
 import numpy as np
 import matplotlib.pyplot as plt
 from mpl_toolkits.axes_grid1 import make_axes_locatable
@@ -148,7 +197,7 @@ def plotVal( epoch=0,*datasets, title=["input","prediction","truth"], saveDir="p
     plt.savefig(filename)
     plt.close(fig) 
 
-    """   使用方法
+    """    
         if total >= 3:
             break
     if epoch % 2 == 0:                    
@@ -200,11 +249,11 @@ def plotValplus(epoch=0, *datasets, title=["input", "prediction", "truth"], save
     filename = f"{saveDir}/epoch_{epoch}.png"
     plt.savefig(filename)
     plt.close(fig)
-    """   使用方法
+    """    
     if total >= 3:
         break
     if epoch % 2 == 0: 
-       plotVal(epoch,    
+       plotValplus(epoch,    
            data[0][0].detach().cpu().numpy().squeeze(),  # 使用 detach()
            output[0][0].detach().cpu().numpy().squeeze(),  # 使用 detach()
            label[0][0].detach().cpu().numpy().squeeze(),  # 使用 detach()
@@ -213,30 +262,6 @@ def plotValplus(epoch=0, *datasets, title=["input", "prediction", "truth"], save
        )
     """
 
-import numpy as np
-import matplotlib
-matplotlib.use("Agg") 
-import matplotlib.pyplot as plt
-from mpl_toolkits.axes_grid1 import make_axes_locatable
-from hjnwtx.colormap import cmp_hjnwtx
-from shancx import crDir
-import os
-def plot_dataset(ax, data, title, cmap, vmin, vmax):
-    """
-    Helper function to plot a single dataset on a given axis.
-    """
-    im = ax.matshow(data, cmap=cmap, vmin=vmin, vmax=vmax)    
-    # Remove axis ticks
-    ax.set_xticks([])
-    ax.set_yticks([])    
-    # Add colorbar
-    divider = make_axes_locatable(ax)
-    cax = divider.append_axes("right", size="5%", pad=0.05)
-    cbar = plt.colorbar(im, cax=cax, ticks=np.linspace(vmin, vmax, 15))
-    cbar.set_ticks(np.linspace(vmin, vmax, 15))    
-    # Set title
-    ax.set_title(title)    
-    return im
 
 def plotValplus1(epoch=0, *datasets, title=["input", "prediction", "truth"], saveDir="plotValplus", cmap='summer'):
     """
@@ -270,6 +295,43 @@ def plotValplus1(epoch=0, *datasets, title=["input", "prediction", "truth"], sav
     plt.savefig(filename)
     plt.close(fig) 
 
+    """    
+    if total >= 3:
+        break
+    if epoch % 2 == 0: 
+       plotValplus1(epoch,    
+           data[0][0].detach().cpu().numpy().squeeze(),  # 使用 detach()
+           output[0][0].detach().cpu().numpy().squeeze(),  # 使用 detach()
+           label[0][0].detach().cpu().numpy().squeeze(),  # 使用 detach()
+           title=["input", "prediction", "groundtruth"], 
+           saveDir="plot_train_dir"
+       )
+    """
+
+import numpy as np
+import matplotlib
+matplotlib.use("Agg") 
+import matplotlib.pyplot as plt
+from mpl_toolkits.axes_grid1 import make_axes_locatable
+from hjnwtx.colormap import cmp_hjnwtx
+from shancx import crDir
+import os
+def plot_dataset(ax, data, title, cmap, vmin, vmax):    #Cited methods
+    """
+    Helper function to plot a single dataset on a given axis.
+    """
+    im = ax.matshow(data, cmap=cmap, vmin=vmin, vmax=vmax)    
+    # Remove axis ticks
+    ax.set_xticks([])
+    ax.set_yticks([])    
+    # Add colorbar
+    divider = make_axes_locatable(ax)
+    cax = divider.append_axes("right", size="5%", pad=0.05)
+    cbar = plt.colorbar(im, cax=cax, ticks=np.linspace(vmin, vmax, 15))
+    cbar.set_ticks(np.linspace(vmin, vmax, 15))    
+    # Set title
+    ax.set_title(title)    
+    return im
 
 import matplotlib.pyplot as plt
 from mpl_toolkits.axes_grid1 import make_axes_locatable
@@ -285,27 +347,24 @@ def calculate_colorbar_range(data):
     vmin = int(np.nanmin(data))
     vmax = int(np.nanmax(data))
     return vmin, vmax
-def plot_grid_data(data, titles=None, save_dir="plots", name="temp",
-                   cmap="viridis", vmin=None, vmax=None):
+def plotgriddata(data, titles=None,name="temp", save_dir="plots", 
+                   cmap="viridis", vmin=None, vmax=None):   #Cited methods
     if not isinstance(data, np.ndarray) or data.ndim != 3:
-        raise ValueError("输入数据必须为三维numpy数组 [num_images, height, width]")    
+        raise ValueError("The input data must be a three-dimensional NumPy array [num_images, height, width]")    
     num_images = data.shape[0]
     titles = titles or [f"Data {i}" for i in range(num_images)]
-    if vmin is None or vmax is None:
-        vmin, vmax = calculate_colorbar_range(data)
     ncols = int(np.ceil(np.sqrt(num_images)))
     nrows = int(np.ceil(num_images / ncols))
     fig, axes = plt.subplots(nrows, ncols, figsize=(ncols * 3, nrows * 3))
     axes = axes.ravel()
     for i in range(num_images):
         ax = axes[i]
-        im = ax.imshow(data[i], cmap=cmap, vmin=vmin, vmax=vmax)
+        im = ax.imshow(data[i], cmap=cmap)
         ax.set_title(titles[i])
         ax.axis('off')        
         divider = make_axes_locatable(ax)
         cax = divider.append_axes("right", size="5%", pad=0.05)
         cbar = plt.colorbar(im, cax=cax,
-                          ticks=np.linspace(vmin, vmax, 15),  
                           format='%.1f')   
         cbar.ax.tick_params(labelsize=6)   
     for j in range(num_images, len(axes)):
@@ -316,7 +375,7 @@ def plot_grid_data(data, titles=None, save_dir="plots", name="temp",
     filename = f"{name}_{timestamp}.png"
     plt.savefig(os.path.join(save_dir, filename), dpi=300)
     plt.close()
-def drawpic_com(basedata, save_dir="plots", name="temp", cmap="summer"): 
+def plotDrawpic(basedata, save_dir="plotDrawpic_com", name="temp", cmap="summer"): 
     data_all = basedata[:,::2,::2]
     if isinstance(name, str):
         print("name str")
@@ -324,7 +383,7 @@ def drawpic_com(basedata, save_dir="plots", name="temp", cmap="summer"):
     else:
         titles = [f"{i}" for i in name.strftime("%Y%m%d%H%M%S")]
         name = name.strftime("%Y%m%d%H%M%S")[0]
-    plot_grid_data(
+    plotgriddata(
         data=data_all,
         titles=titles,
         name=name,
@@ -333,398 +392,65 @@ def drawpic_com(basedata, save_dir="plots", name="temp", cmap="summer"):
     ) 
 
 """
-drawpic_com(Data_con, save_dir="plots_H9", name=timeList )
-"""
-
-import os
-import matplotlib.pyplot as plt
-from matplotlib.ticker import MaxNLocator
-class trainingVis:
-    def __init__(self, args=None, dataset_key_map=None, root_path="./"):
-        self.args = args
-        self.dataset_key_map = dataset_key_map
-        self.root_path = root_path
-        self.record = {
-            "train_loss": [],
-            "train_psnr": [],
-            "val_loss": [],
-            "val_psnr": [],
-        }
-        self.x_epoch = []
-        self.output_dir = self._get_output_dir()
-    def _get_output_dir(self):
-        """生成输出目录路径并创建目录"""
-        output_dir = os.path.join(
-            self.root_path,
-            "Rec",
-            # "weights_dir",
-            # self.dataset_key_map[self.args.dataset_key],
-            "trainvalViscure",
-        )
-        os.makedirs(output_dir, exist_ok=True)
-        return output_dir
-    def _plot_curve(self, ax, x, y_train, y_val, y_label, train_color="blue", val_color="red"):
-        """绘制单条曲线并优化坐标轴显示"""
-        if y_train is not None:
-            ax.plot(x, y_train, marker='o', linestyle='-', color=train_color, label="Train")
-        if y_val is not None:
-            ax.plot(x, y_val, marker='o', linestyle='-', color=val_color, label="Val")        
-        # 设置坐标轴标签和格式
-        ax.set_xlabel("Epoch", fontsize=10)
-        ax.set_ylabel(y_label, fontsize=10)
-        ax.set_title(f"{y_label} Curve", fontsize=12)        
-        # 配置x轴刻度（确保整数显示）
-        ax.xaxis.set_major_locator(MaxNLocator(integer=True))
-        plt.setp(ax.get_xticklabels(), rotation=30, ha='right', fontsize=8)        
-        # 配置y轴刻度
-        ax.yaxis.set_major_locator(MaxNLocator(nbins=6))
-        plt.setp(ax.get_yticklabels(), fontsize=8)        
-        # 添加图例
-        ax.legend(loc="upper right", fontsize=8)
-    def draw_curve(self, epoch=None, train_loss=None, train_psnr=None, val_loss=None, val_psnr=None):
-        """动态绘制训练曲线并根据数据存在性调整布局"""
-        self.record["train_loss"].append(train_loss)
-        self.record["val_loss"].append(val_loss)
-        self.x_epoch.append(epoch)        
-        has_psnr = train_psnr is not None and val_psnr is not None
-        if has_psnr:
-            self.record["train_psnr"].append(train_psnr)
-            self.record["val_psnr"].append(val_psnr)
-        else:
-            self.record["train_psnr"].append(None)
-            self.record["val_psnr"].append(None)
-        fig = plt.figure(figsize=(10, 4.5) if has_psnr else (6, 4.5))
-        plt.subplots_adjust(wspace=0.3 if has_psnr else 0)        
-        ax0 = fig.add_subplot(111 if not has_psnr else 121)        
-        # 如果只有train_loss数据存在
-        if train_loss is not None and val_loss is None:
-            self._plot_curve(ax0
-                             ,self.x_epoch
-                             ,self.record["train_loss"]
-                             ,None
-                             ,"Loss"
-                             ,train_color="blue"
-                             ,val_color="red"
-                             )        
-        elif val_loss is not None and train_loss is None:
-            self._plot_curve(ax0, 
-                             self.x_epoch, 
-                             None, 
-                             self.record["val_loss"], 
-                             "Loss", 
-                             train_color="blue", 
-                             val_color="red"
-                             )
-        else:
-            self._plot_curve(ax0, 
-                             self.x_epoch, 
-                             self.record["train_loss"], 
-                             self.record["val_loss"], 
-                             "Loss", train_color="blue", 
-                             val_color="red"
-                             )
-        if has_psnr:
-            ax1 = fig.add_subplot(122)
-            self._plot_curve(ax1, 
-                             self.x_epoch, 
-                             self.record["train_psnr"], 
-                             self.record["val_psnr"], 
-                             "PSNR", 
-                             train_color="orange", 
-                             val_color="grey"
-                             )
-        plt.tight_layout()
-        fig.savefig(
-            os.path.join(self.output_dir, f"train_{epoch}.jpg"),
-            dpi=300,
-            bbox_inches="tight",
-            pad_inches=0.1
-        )
-        plt.close(fig)
-        """
-        vis3 = TrainingVis(root_path="./Rec3")
-        vis4 = TrainingVis(root_path="./Rec4")
-        vis5 = TrainingVis(root_path="./Rec5")
-        for epoch in range(t.epoch, t.epoch + args.num_epochs):
-            train_loss, train_psnr = t.train(epoch)
-            val_loss, val_psnr = t.val(epoch)
-            if (epoch + 1) % 3 == 0:
-                # t.draw_curve(fig, epoch, train_loss, train_psnr, val_loss, val_psnr)
-                vis.draw_curve(epoch, train_loss, train_psnr, val_loss, val_psnr)
-                vis1.draw_curve(epoch, train_loss,val_loss)
-                vis2.draw_curve(epoch, train_loss,val_loss,train_psnr,val_psnr)
-                vis3.draw_curve(epoch, train_loss,val_loss,train_psnr,val_psnr)
-                vis4.draw_curve(epoch, train_loss)
-                vis5.draw_curve(epoch, val_loss)
-        ------------------------
-        from shancx.Plot import trainingVis
-        vis= trainingVis(root_path="./Rec3")  
-        if (epoch + 1) % 3== 0:     
-            vis.draw_curve(epoch, epoch_loss.detach().cpu().numpy(),epoch_val_loss.detach().cpu().numpy())
-        """
-
-class trainingVisplus:
-    def __init__(self, args=None, dataset_key_map=None, root_path="./"):
-        self.args = args
-        self.dataset_key_map = dataset_key_map
-        self.root_path = root_path
-        self.record = {
-            "train_loss": [],
-            "train_psnr": [],
-            "train_acc": [],
-            "val_loss": [],
-            "val_psnr": [],
-            "val_acc": [],
-        }
-        self.x_epoch = []
-        self.output_dir = self._get_output_dir()
-
-    def _get_output_dir(self):
-        """生成输出目录路径并创建目录"""
-        output_dir = os.path.join(
-                                  self.root_path,
-                                  "Rec",
-                                  "trainvalViscure",
-                                 )
-        os.makedirs(output_dir, exist_ok=True)
-        return output_dir
-
-    def _plot_curve(
-                    self, 
-                    ax, 
-                    x, 
-                    y_train, 
-                    y_val, 
-                    y_label, 
-                    train_color="blue", 
-                    val_color="red"
-                    ):
-        """绘制单条曲线并优化坐标轴显示"""
-        if y_train is not None:
-            ax.plot(x, y_train, 
-                    marker='o', 
-                    linestyle='-', 
-                    color=train_color, 
-                    label="Train"
-                    )
-        if y_val is not None:
-            ax.plot(x, y_val, 
-                    marker='o', 
-                    linestyle='-', 
-                    color=val_color, 
-                    label="Val"
-                    )
-        # 设置坐标轴标签和格式
-        ax.set_xlabel("Epoch", fontsize=10)
-        ax.set_ylabel(y_label, fontsize=10)
-        ax.set_title(f"{y_label} Curve", fontsize=12)
-        # 配置x轴刻度（确保整数显示）
-        ax.xaxis.set_major_locator(MaxNLocator(integer=True))
-        plt.setp(ax.get_xticklabels(), rotation=30, ha='right', fontsize=8)
-        # 配置y轴刻度
-        ax.yaxis.set_major_locator(MaxNLocator(nbins=6))
-        plt.setp(ax.get_yticklabels(), fontsize=8)
-        # 添加图例
-        ax.legend(loc="upper right", fontsize=8)
-
-    def draw_curve(
-                   self, epoch=None, 
-                   train_loss=None, 
-                   train_psnr=None, 
-                   val_loss=None, 
-                   val_psnr=None, 
-                   train_acc=None, 
-                   val_acc=None
-                   ):
-        """动态绘制训练曲线并根据数据存在性调整布局"""
-        # 更新训练记录
-        self.record["train_loss"].append(train_loss)
-        self.record["val_loss"].append(val_loss)
-        self.x_epoch.append(epoch)
-
-        # 有条件地更新PSNR和Acc记录
-        has_psnr = train_psnr is not None and val_psnr is not None
-        has_acc = train_acc is not None and val_acc is not None
-
-        if has_psnr:
-            self.record["train_psnr"].append(train_psnr)
-            self.record["val_psnr"].append(val_psnr)
-        else:
-            # 用None占位保持数据对齐
-            self.record["train_psnr"].append(None)
-            self.record["val_psnr"].append(None)
-
-        if has_acc:
-            self.record["train_acc"].append(train_acc)
-            self.record["val_acc"].append(val_acc)
-        else:
-            # 用None占位保持数据对齐
-            self.record["train_acc"].append(None)
-            self.record["val_acc"].append(None)
-
-        # 创建自适应布局的画布
-        num_plots = 1 + int(has_psnr) + int(has_acc)
-        fig_width = 4 * num_plots  # 每个子图宽度为4
-        fig = plt.figure(figsize=(fig_width, 4.5))
-        plt.subplots_adjust(wspace=0.3)
-
-        # 绘制Loss曲线
-        ax0 = fig.add_subplot(1, num_plots, 1)
-        self._plot_curve(
-                         ax0, 
-                         self.x_epoch, 
-                         self.record["train_loss"], 
-                         self.record["val_loss"], 
-                         "Loss", 
-                         train_color="blue", 
-                         val_color="red"
-                         )
-
-        # 绘制PSNR曲线（如果存在）
-        if has_psnr:
-            ax1 = fig.add_subplot(1, num_plots, 2)
-            self._plot_curve(
-                             ax1, 
-                             self.x_epoch, 
-                             self.record["train_psnr"], 
-                             self.record["val_psnr"], 
-                             "PSNR", 
-                             train_color="orange", 
-                             val_color="grey"
-                             )
-
-        # 绘制Acc曲线（如果存在）
-        if has_acc:
-            ax2 = fig.add_subplot(1, num_plots, num_plots)
-            self._plot_curve(
-                             ax2, 
-                             self.x_epoch, 
-                             self.record["train_acc"], 
-                             self.record["val_acc"],
-                             "Accuracy", 
-                             train_color="grey", 
-                             val_color="purple"
-                             )
-
-        # 优化布局并保存
-        plt.tight_layout()
-        fig.savefig(
-            os.path.join(self.output_dir, f"train_{epoch}.jpg"),
-            dpi=300,
-            bbox_inches="tight",
-            pad_inches=0.1
-        )
-        plt.close(fig)
-        """
-
-        vis1= trainingVisplus(root_path="./Rec4") 
-        for epoch in range(t.epoch, t.epoch + args.num_epochs):
-            train_loss, train_psnr = t.train(epoch)
-            val_loss, val_psnr = t.val(epoch)
-            if (epoch + 1) % 3 == 0:
-            vis1.draw_curve(epoch=epoch, 
-                            train_loss=epoch_loss.detach().cpu().numpy(), 
-                            val_loss=epoch_val_loss.detach().cpu().numpy(),  
-                            train_acc=epoch_accuracy, 
-                            val_acc=epoch_val_accuracy
-                            )
-
-        """
-
-# @staticmethod
-# def calculate_psnr(img1, img2):
-#     return 10. * torch.log10(1. / torch.mean((img1 - img2) ** 2))   
-    """  使用方法
-    psnr += self.calculate_psnr(fake_img, label).item()
-    total += 1
-     mean_psnr = psnr / total
-    """ 
-
+drawpic_com(Data_con, save_dir="plotDrawpic_com", name=timeList )
+""" 
 from hjnwtx.colormap import cmp_hjnwtx
 import matplotlib.pyplot as plt
 from mpl_toolkits.axes_grid1 import make_axes_locatable
 import numpy as np
 import os
 import datetime
-def calculate_colorbar_range(data):
-    """计算色标范围"""
-    valid_data = data[~np.isnan(data)]
-    vmin = int(np.min(valid_data))
-    vmax = int(np.max(valid_data))
- 
-    return vmin, vmax
-def plot_grid_data(data, titles=None, saveDir="plots", name="temp",
-                   cmap="viridis", vmin=None, vmax=None):
+def plot_grid_data(data, titles=None, saveDir="plots", name="temp", cmap="summer",radarnmc=1):
     if not isinstance(data, np.ndarray) or data.ndim != 3:
-        raise ValueError("输入数据必须为三维numpy数组 [num_images, height, width]")    
+        raise ValueError("The input data must be a three-dimensional NumPy array [num_images, height, width]")    
     num_images = data.shape[0]
     titles = titles or [f"Data {i}" for i in range(num_images)]    
-    # 计算色标范围
-    if vmin is None or vmax is None:
-        vmin, vmax = calculate_colorbar_range(data)    
-    # 计算子图布局
     ncols = int(np.ceil(np.sqrt(num_images)))
     nrows = int(np.ceil(num_images / ncols))    
-    # 创建子图
-    fig, axes = plt.subplots(nrows, ncols, figsize=(ncols * 3, nrows * 3))
-    axes = axes.ravel()
+    fig, axes = plt.subplots(nrows, ncols, figsize=(ncols * 4, nrows * 3))
+    if num_images == 1:
+        axes = np.array([[axes]])
+    elif axes.ndim == 1:
+        axes = axes.reshape(1, -1)    
+    axes_flat = axes.ravel()    
     for i in range(num_images):
-        if i==int(num_images-1):
-            vmin = 0
-            vmax = 70
-            ax = axes[i]
-            im = ax.imshow(data[i], cmap=cmp_hjnwtx["radar_nmc"], vmin=vmin, vmax=vmax) #cmp_hjnwtx["radar_nmc"]
+        ax = axes_flat[i]
+        if i >= num_images - radarnmc:
+            im = ax.imshow(data[i], cmap=cmp_hjnwtx["radar_nmc"])
         else:
-            vmin = 190
-            vmax = 300
-            ax = axes[i]
-            im = ax.imshow(data[i], cmap=cmap, vmin=vmin, vmax=vmax)
+            im = ax.imshow(data[i], cmap=cmap)        
         ax.set_title(titles[i])
-        ax.axis('off')        
-        # 添加色标
+        ax.axis('off')
         divider = make_axes_locatable(ax)
         cax = divider.append_axes("right", size="5%", pad=0.05)
-        cbar = plt.colorbar(im, cax=cax,
-                          ticks=np.linspace(vmin, vmax, 15),  # 增加刻度密度
-                          format='%.1f')  # 设置数值格式
-        cbar.ax.tick_params(labelsize=6)  # 调整刻度文字大小    
-    # 隐藏空子图
-    for j in range(num_images, len(axes)):
-        axes[j].axis('off')    
-    # 保存图像
+        plt.colorbar(im, cax=cax)
+    for j in range(num_images, len(axes_flat)):
+        axes_flat[j].axis('off')    
     plt.tight_layout()
     os.makedirs(saveDir, exist_ok=True)
     timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
     filename = f"{name}_{timestamp}.png"
     plt.savefig(os.path.join(saveDir, filename), dpi=300)
     plt.close()
-def plotTr(base_up, base_down, name="plotTr", saveDir="plotTr",  cmap="summer"): #viridis
-    """
-    组合数据并调用绘图函数    
-    Args:
-        base_up: 上部数据数组 [shape_len, H, W]
-        base_down: 下部数据数组 [1, H, W]
-        shape_len: 上部数据数量
-        name: 输出文件名前缀
-        cmap: 颜色映射
-    """  
-    # 合并数据并生成标题
+
+def plotTr(base_up, base_down, name="plotTr", saveDir="plotTr",cmap="summer",radarnmc=1):
     data_all = np.concatenate([base_up, base_down], axis=0)
-    titles = [f"B_{i}" for i in range(base_up.shape[0])] + [f"radar_{i+1}" for i in range(base_down.shape[0])]    
-    # 调用绘图函数
+    titles = [f"Pic_{i}" for i in range(base_up.shape[0])] + [f"Pic_{i+1}" for i in range(base_down.shape[0])]    
     plot_grid_data(
         data=data_all,
         titles=titles,
         name=name,
         saveDir=saveDir,
-        cmap=cmap
-    ) 
-    """ 
-    if __name__ == "__main__":
-        base_up = np.random.rand(10, 50, 50) * 70
-        base_down = np.random.rand(1, 50, 50) * 70
-        plotTr(base_up, base_down, name="radar_plot") #   radar_mask.detach().cpu().numpy()  tensor转numpy 
-    """
+        cmap=cmap,
+        radarnmc = radarnmc
+    )
+
+""" 
+if __name__ == "__main__":
+    base_up = np.random.rand(10, 50, 50) * 70
+    base_down = np.random.rand(1, 50, 50) * 70
+    plotTr(base_up, base_down, name="radar_plot",cmap="summer",radarnmc=1) #   radar_mask.detach().cpu().numpy()  tensor转numpy 
+"""
 
 import numpy as np
 import matplotlib.pyplot as plt