Changeset 6508 for TOOLS/WATER_BUDGET/lmdz.py
- Timestamp:
- 06/13/23 12:58:38 (13 months ago)
- File:
-
- 1 edited
Legend:
- Unmodified
- Added
- Removed
-
TOOLS/WATER_BUDGET/lmdz.py
r6265 r6508 238 238 if math == xr : 239 239 p2D = xr.DataArray (p2D) 240 p2D.attrs = p1D.attrs 240 for attr in p1D.attrs : 241 p2D.attrs[attr] = p1D.attrs[attr] 241 242 p2D = p2D.rename ( {p2D.dims[0]:p1D.dims[0], p2D.dims[-1]:'x', p2D.dims[-2]:'y'} ) 242 243 243 244 return p2D 244 245 245 def geo2point ( p2D, cumulPoles=False, dim1D=' points_physiques') :246 def geo2point ( p2D, cumulPoles=False, dim1D='cell', debug=False ) : 246 247 ''' 247 248 From 2D (lat, lon) to 1D (points_phyiques) … … 259 260 260 261 jpn = jpi*(jpj-2) + 2 262 263 if debug : 264 print ( f'{jpi=} {jpj=} {jpn=} {jpt=}' ) 261 265 262 266 if jpt == -1 : … … 277 281 if math == xr : 278 282 p1D = xr.DataArray (p1D) 279 p1D.attrs = p2D.attrs 283 for attr in p2D.attrs : 284 p1D.attrs[attr] = p2D.attrs[attr] 280 285 p1D = p1D.rename ( {p1D.dims[0]:p2D.dims[0], p1D.dims[-1]:dim1D} ) 281 286 … … 286 291 return p1D 287 292 288 def geo3point ( p3D, cumulPoles=False, dim1D=' points_physiques' ) :293 def geo3point ( p3D, cumulPoles=False, dim1D='cell' ) : 289 294 ''' 290 295 From 3D (lev, lat, lon) to 2D (lev, points_phyiques) … … 314 319 if math == xr : 315 320 p2D = xr.DataArray (p2D) 316 p2D.attrs = p3D.attrs 321 for attr in p2D.attrs : 322 p2D.attrs[attr] = p3D.attrs[attr] 317 323 p2D = p2D.rename ( {p2D.dims[-1]:dim1D, p2D.dims[-2]:p3D.dims[-3]} ) 318 324
Note: See TracChangeset
for help on using the changeset viewer.