shancx 1.9.33.224__py3-none-any.whl → 1.9.33.227__py3-none-any.whl

shancx/Read.py

@@ -1,9 +1,84 @@
+
+import xarray as xr
+import numpy as np
+import gzip
+import tempfile
+import os
+backend_kwargs = {
+    'errors': 'ignore',  
+    # 'errors': 'raise'   
+    # 'errors': 'warn'    
+}
+def getGrib(filepath):
+    tmp_path = filepath 
+    if filepath.endswith('.gz'):
+        with gzip.open(filepath, 'rb') as f:
+            with tempfile.NamedTemporaryFile(suffix='.grib2', delete=False) as tmp:
+                tmp.write(f.read())
+                tmp_path = tmp.name   
+    try:
+        ds = xr.open_dataset(tmp_path, engine='cfgrib', backend_kwargs=backend_kwargs)
+        var_name = list(ds.data_vars)[0]
+        data_var = ds[var_name] 
+        data_values = np.where(data_var.values < -900, np.nan, data_var.values)
+        if 'latitude' in ds.coords and 'longitude' in ds.coords:
+            lats = ds.latitude.values
+            lons = ds.longitude.values
+        elif 'lat' in ds.coords and 'lon' in ds.coords:
+            lats = ds.lat.values
+            lons = ds.lon.values
+        elif hasattr(ds, 'latitude') and hasattr(ds, 'longitude'):
+            lats = ds.latitude.values
+            lons = ds.longitude.values
+        else:
+            for var_name in ds.data_vars:
+                var = ds[var_name]
+                if hasattr(var, 'latitude') and hasattr(var, 'longitude'):
+                    lats = var.latitude.values
+                    lons = var.longitude.values
+                    break
+            else:
+                raise ValueError("未找到经纬度坐标")       
+        return {
+            'data': data_values,
+            'lats': lats,
+            'lons': lons 
+        }        
+    finally:
+        os.unlink(tmp_path)
+
+"""
+input_Path = './Composite_00.50_20250701-010044.grib2.gz'
+result = getGrib(input_Path)
+"""
+
 import pygrib
 import numpy as np
 import pandas as pd
-def readGrib(file_path, target_param=None):
+import gzip
+import tempfile
+def readGrib(filepath, target_param=None):
+    filepath= filepath  
+    if filepath.endswith('.gz'):
+        with gzip.open(filepath, 'rb') as f:
+            with tempfile.NamedTemporaryFile(suffix='.grib2', delete=False) as tmp:
+                tmp.write(f.read())
+                filepath = tmp.name  
     try:
-        with pygrib.open(file_path) as grbs:
+        with pygrib.open(filepath) as grbs:         
+            if target_param:
+                try:
+                    grb = grbs.select(shortName=target_param)[0]
+                except:
+                    try:
+                        grb = grbs.select(parameterName=target_param)[0]
+                    except:
+                        raise ValueError(f"未找到参数: {target_param}")
+            else:
+                grb = grbs[1]
+            data = grb.values
+            lats, lons = grb.latlons() 
+            #data2 
             field_info = []
             for grb in grbs:
                 field_info.append({
@@ -16,18 +91,6 @@ def readGrib(file_path, target_param=None):
                                    'units': getattr(grb, 'units', 'N/A'),
                                    'shape': grb.values.shape
                                  })             
-            if target_param:
-                try:
-                    grb = grbs.select(shortName=target_param)[0]
-                except:
-                    try:
-                        grb = grbs.select(parameterName=target_param)[0]
-                    except:
-                        raise ValueError(f"未找到参数: {target_param}")
-            else:
-                grb = grbs[1]
-            data = grb.values
-            lats, lons = grb.latlons()            
             return {
                 'data': data,
                 'lats': lats,
@@ -36,12 +99,14 @@ def readGrib(file_path, target_param=None):
                     'parameterName': grb.parameterName,
                     'level': grb.level,
                     'validDate': grb.validDate,
-                    'units': grb.units
+                    'units': grb.units,
+                    "field_info":field_info                    
                 }
             }            
     except Exception as e:
         print(f"GRIB读取错误: {str(e)}")
         return None
+"""
 if __name__ == "__main__":
     path = "/mnt/wtx_weather_forecast/CMA_DATA/NAFP/EC/C1D/2024/2024112720/ECMFC1D_PRTY_1_2024112712_GLB_1_2.grib2"
     result = readGrib(path)    
@@ -50,9 +115,6 @@ if __name__ == "__main__":
         print("经度范围:", np.min(result['lons']), "~", np.max(result['lons']))
         print("纬度范围:", np.min(result['lats']), "~", np.max(result['lats']))
         print("参数单位:", result['metadata']['units'])
-        """
-        latArr = latMat[:,0]
-        lonArr = lonMat[0]
-        """    
+"""
+  
 
- 

{shancx-1.9.33.224.dist-info → shancx-1.9.33.227.dist-info}/METADATA

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

{shancx-1.9.33.224.dist-info → shancx-1.9.33.227.dist-info}/RECORD

@@ -5,7 +5,7 @@ shancx/Lib.py,sha256=GUAspllSxk39mvj-F1Q8Ys0EcY_lQfZPRGPE7L3x4SE,977
 shancx/Path1.py,sha256=vX4A5RgdwVyIVVNZRocl18rOu1Z8YMLfDb0B92-0bE8,5334
 shancx/Point.py,sha256=gyIomOVbNoz6SOcAhhTS26NHBeJ0TOwB-ljNsBWN1ZE,1909
 shancx/QC.py,sha256=MFxbPMEwrnk5l0sTFMrx13wPlXJ5WI5TovzaiBoEr0E,9848
-shancx/Read.py,sha256=qaX4Qf9J8vXjam60PWjWdAL-1hvEwcEe_d4bR4MZ2OU,2465
+shancx/Read.py,sha256=94FNUWiZn71X6ArZhyqhyORNcWJffC7Hz76BAtF9CJc,4577
 shancx/Resize.py,sha256=uwiOhD-aWPDLvIwOYEX-XXN_suOXv6hPDcaGyr4JLC4,4044
 shancx/ZR.py,sha256=5APchqVoI1G2MkrN9YJSWCwfLUygcSFIsUXSo9aq1Qg,341
 shancx/__init__.py,sha256=6sfiHElECqfy5eEEGqaCAw4YiKxgKfJ5IFFh9yvtewk,11442
@@ -87,7 +87,7 @@ shancx/Train/multiGpu.py,sha256=D_oZeiSc7VWktpnVDwrFOC1CYZSt9rxOKY5lngE5vFg,820
 shancx/Train/prepare.py,sha256=vL_8UOA66oZCBIwCICtihsGibivtNgaVJGulJxfNdn8,6793
 shancx/Train/renet50.py,sha256=wEhYk1X96WE5zuqHqVxWLJa-A5jDNkz4z6edORNufnA,6428
 shancx/tensBoard/__init__.py,sha256=ga2C5YyJITvvQA1ocpxna_KNFnNRJVwkTjLoIglLZUQ,993
-shancx-1.9.33.224.dist-info/METADATA,sha256=4578s6UvEQqzaWNvvxbHegy-Fab_bjob6EyGFe08-s0,850
-shancx-1.9.33.224.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-shancx-1.9.33.224.dist-info/top_level.txt,sha256=akfCS1vKWz3pNmEN_yN9ZiGp-60IQY5ET38mRx_i_-4,7
-shancx-1.9.33.224.dist-info/RECORD,,
+shancx-1.9.33.227.dist-info/METADATA,sha256=2IF7Rl9DDL9BPZypQ0PtDTid3CfYF7wOu9HjWhQO42Y,850
+shancx-1.9.33.227.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+shancx-1.9.33.227.dist-info/top_level.txt,sha256=akfCS1vKWz3pNmEN_yN9ZiGp-60IQY5ET38mRx_i_-4,7
+shancx-1.9.33.227.dist-info/RECORD,,