shancx 1.9.33.224__tar.gz → 1.9.33.226__tar.gz

{shancx-1.9.33.224 → shancx-1.9.33.226}/PKG-INFO

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

{shancx-1.9.33.224 → shancx-1.9.33.226}/setup.py

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

shancx-1.9.33.226/shancx/Read.py

@@ -0,0 +1,111 @@
+import xarray as xr
+import numpy as np
+import gzip
+import tempfile
+import os
+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')
+        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("未找到经纬度坐标")
+        lat_range = (float(np.nanmin(lats)), float(np.nanmax(lats)))
+        lon_range = (float(np.nanmin(lons)), float(np.nanmax(lons)))        
+        return {
+            'data': data,
+            '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):
+    try:
+        with pygrib.open(file_path) as grbs:    
+            file_path= file_path  
+            if filepath.endswith('.gz'):
+                with gzip.open(filepath, 'rb') as f:
+                    with tempfile.NamedTemporaryFile(suffix='.grib2', delete=False) as tmp:
+                        tmp.write(f.read())
+                        file_path = tmp.name       
+            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({
+                                   'messageNumber': grb.messagenumber,
+                                   'parameterName': getattr(grb, 'parameterName', 'N/A'),
+                                   'shortName': getattr(grb, 'shortName', 'N/A'),
+                                   'level': getattr(grb, 'level', -999),
+                                   'typeOfLevel': getattr(grb, 'typeOfLevel', 'N/A'),
+                                   'validDate': getattr(grb, 'validDate', 'N/A'),
+                                   'units': getattr(grb, 'units', 'N/A'),
+                                   'shape': grb.values.shape
+                                 })             
+            return {
+                'data': data,
+                'lats': lats,
+                'lons': lons,
+                'metadata': {
+                    'parameterName': grb.parameterName,
+                    'level': grb.level,
+                    'validDate': grb.validDate,
+                    '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)    
+    if result:
+        print("\n数据矩阵形状:", result['data'].shape)
+        print("经度范围:", np.min(result['lons']), "~", np.max(result['lons']))
+        print("纬度范围:", np.min(result['lats']), "~", np.max(result['lats']))
+        print("参数单位:", result['metadata']['units'])
+  
+

{shancx-1.9.33.224 → shancx-1.9.33.226}/shancx.egg-info/PKG-INFO

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

shancx-1.9.33.224/shancx/Read.py

@@ -1,58 +0,0 @@
-import pygrib
-import numpy as np
-import pandas as pd
-def readGrib(file_path, target_param=None):
-    try:
-        with pygrib.open(file_path) as grbs:
-            field_info = []
-            for grb in grbs:
-                field_info.append({
-                                   'messageNumber': grb.messagenumber,
-                                   'parameterName': getattr(grb, 'parameterName', 'N/A'),
-                                   'shortName': getattr(grb, 'shortName', 'N/A'),
-                                   'level': getattr(grb, 'level', -999),
-                                   'typeOfLevel': getattr(grb, 'typeOfLevel', 'N/A'),
-                                   'validDate': getattr(grb, 'validDate', 'N/A'),
-                                   '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,
-                'lons': lons,
-                'metadata': {
-                    'parameterName': grb.parameterName,
-                    'level': grb.level,
-                    'validDate': grb.validDate,
-                    'units': grb.units
-                }
-            }            
-    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)    
-    if result:
-        print("\n数据矩阵形状:", result['data'].shape)
-        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]
-        """    
-
-