shancx 1.9.33.229__tar.gz → 1.9.33.230__tar.gz

{shancx-1.9.33.229 → shancx-1.9.33.230}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: shancx
-Version: 1.9.33.229
+Version: 1.9.33.230
 Summary: A simple timer decorator
 Home-page: https://gitee.com/shancx
 Author: shancx

{shancx-1.9.33.229 → shancx-1.9.33.230}/setup.py

@@ -3,7 +3,7 @@
 from setuptools import setup, find_packages
 setup(
     name="shancx",
-    version="1.9.33.229",
+    version="1.9.33.230",
     packages=find_packages(),
     description="A simple timer decorator",
     long_description=open("README.md").read(),

{shancx-1.9.33.229 → shancx-1.9.33.230}/shancx/Plot/__init__.py

@@ -394,13 +394,13 @@ def plotDrawpic(basedata, save_dir="plotDrawpic_com", name="temp", cmap="summer"
 """
 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 plot_grid_data(data, titles=None, saveDir="plots", name="temp", cmap="summer",radarnmc=1):
+from hjnwtx.colormap import cmp_hjnwtx
+def plot_grid_data(data, titles=None, saveDir="plots", name="temp", cmap="summer",radarnmc=1,minMax=True):
     if not isinstance(data, np.ndarray) or data.ndim != 3:
         raise ValueError("The input data must be a three-dimensional NumPy array [num_images, height, width]")    
     num_images = data.shape[0]
@@ -416,12 +416,12 @@ def plot_grid_data(data, titles=None, saveDir="plots", name="temp", cmap="summer
     for i in range(num_images):
         ax = axes_flat[i]
         if i >= num_images - radarnmc:
-            im = ax.imshow(data[i], cmap=cmp_hjnwtx["radar_nmc"])
+            im = ax.imshow(data[i], cmap=cmp_hjnwtx["radar_nmc"],vmin=0,vmax=70 ) if minMax  else ax.imshow(data[i], cmap=cmp_hjnwtx["radar_nmc"])
         else:
-            im = ax.imshow(data[i], cmap=cmap)        
+            im = ax.imshow(data[i],cmap=cmap,vmin=180,vmax=310 ) if minMax else ax.imshow(data[i])       
         ax.set_title(titles[i])
         ax.axis('off')
-        divider = make_axes_locatable(ax)
+        divider = make_axes_locatable(ax) 
         cax = divider.append_axes("right", size="5%", pad=0.05)
         plt.colorbar(im, cax=cax)
     for j in range(num_images, len(axes_flat)):
@@ -433,7 +433,7 @@ def plot_grid_data(data, titles=None, saveDir="plots", name="temp", cmap="summer
     plt.savefig(os.path.join(saveDir, filename), dpi=300)
     plt.close()
 
-def plotTr(base_up, base_down, name="plotTr", saveDir="plotTr",cmap="summer",radarnmc=1):
+def plotTr(base_up, base_down, name="plotTr", saveDir="plotTr",cmap="summer",radarnmc=1,minMax = True):
     data_all = np.concatenate([base_up, base_down], axis=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(
@@ -442,14 +442,15 @@ def plotTr(base_up, base_down, name="plotTr", saveDir="plotTr",cmap="summer",rad
         name=name,
         saveDir=saveDir,
         cmap=cmap,
-        radarnmc = radarnmc
+        radarnmc = radarnmc,
+        minMax = minMax
     )
 
 """ 
 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 
+    plotTr(base_up, base_down, name="radar_plot",radarnmc=1) #   radar_mask.detach().cpu().numpy()  tensor转numpy 
 """
 
 import numpy as np

{shancx-1.9.33.229 → shancx-1.9.33.230}/shancx.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: shancx
-Version: 1.9.33.229
+Version: 1.9.33.230
 Summary: A simple timer decorator
 Home-page: https://gitee.com/shancx
 Author: shancx