shancx 1.9.33.228__tar.gz → 1.9.33.230__tar.gz

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

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

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

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

{shancx-1.9.33.228 → 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.230/shancx/npz.py

@@ -0,0 +1,29 @@
+import numpy as np
+from pathlib import Path
+import os
+import uuid
+import traceback
+def npzSave(output, data):
+    try:
+        output_path = Path(output)
+        if output_path.suffix in ['.npy', '.nc', '.h5', '.hdf5']:
+            output_path = output_path.with_suffix('.npz')    
+        output_path.parent.mkdir(parents=True, exist_ok=True)
+        temp_filename = output_path.with_suffix(f'.tmp.{uuid.uuid4()}.npz')    
+        np.savez_compressed(temp_filename, data=data)
+        os.replace(temp_filename, output_path)   
+        print(f"{output_path} done")
+    except Exception as e:
+        print(traceback.format_exc())
+        if temp_filename and temp_filename.exists():
+            try:
+                os.remove(temp_filename)
+                print(f"⚠ Temporary files have been cleared: {temp_filename}")
+            except:
+                pass            
+def npzLoad(output_path):
+    return np.load(Path(output_path).with_suffix('.npz'))['data']
+"""
+npzSave(output, data)
+npzLoad(output_path)
+"""

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

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

shancx-1.9.33.228/shancx/npz.py

@@ -1,15 +0,0 @@
-import numpy as np
-from pathlib import Path
-def npzSave(output, data):
-    output = output.replace('.npy', '.npz')
-    output_path = Path(output).with_suffix('.npz')
-    output_path.parent.mkdir(parents=True, exist_ok=True)
-    np.savez_compressed(output_path, data=data)
-    print(f"{output_path} done")
-def npzLoad(output_path):
-    return np.load(Path(output_path).with_suffix('.npz'))['data']
-
-"""
-npzSave(output, data)
-npzLoad(output_path)
-"""