Changeset 960 for trunk/UTIL/PYTHON/planetoplot_v2/examples/easy_get_field.py

Timestamp:

May 15, 2013, 8:29:27 PM (12 years ago)

Author:

aslmd

Message:

UTIL PYTHON planetoplot_v2. MAJOR: created an easy way to load a variable from a netcdf file, see easy_get_field.py [note: now to avoid confusion with .f attributes containing field, .f() method is named .func()] ; MINOR: generic units keyword, quiet mode, dummy xy axis when not given in ppplot, bug fixes for one-value 0D requests.

File:

• trunk/UTIL/PYTHON/planetoplot_v2/examples/easy_get_field.py (modified) (1 diff)

trunk/UTIL/PYTHON/planetoplot_v2/examples/easy_get_field.py

@@ -1 +1 @@
 #! /usr/bin/env python
 from ppclass import pp
-import numpy as np
-import matplotlib.pyplot as mpl
 
+########
+## ppclass allows for getting fields very easily from a netcdf file
+## ... this is contained in the "f" attributes of the pp object
+## ... and the "l" attributes of the pp object contains useful information
+## ... two methods getf() and getfl() are helpful shortcuts
+###########################################
+
+## 1 --> an unique and straightforward request
+##   --> very easy ! see also minimal_field.py
+##############################################
+icetot = pp(file="/home/aymeric/Big_Data/DATAPLOT/diagfired.nc",var="icetot",t=0.5).getf()
+print "icetot", icetot[10,10]
+
+## 2 --> simple multiple request, no labelling
+##############################################
+test = pp(file="/home/aymeric/Big_Data/DATAPLOT/diagfired.nc",t=0.5)
+test.var = ["mtot","icetot"]
+allf = test.getf() # or allf = test.get().f
+##
+mtot = allf[0]
+icetot = allf[1]
+print "mtot", mtot[10,10]
+print "icetot", icetot[10,10]
+
+## 3 --> complex multiple requests and labelling
+################################################
+test = pp(file="/home/aymeric/Big_Data/DATAPLOT/diagfired.nc")
+test.var = ["mtot","icetot"]
+test.t = [0.4,0.5]
+allf,lab = test.getfl()
+##
+icetot04 = allf[lab.index("_v=icetot_t=0.4_")]
+mtot04 = allf[lab.index("_v=mtot_t=0.4_")]
+icetot05 = allf[lab.index("_v=icetot_t=0.5_")]
+mtot05 = allf[lab.index("_v=mtot_t=0.5_")]
+print "mtot04", mtot04[10,10]
+print "icetot04", icetot04[10,10]
+print "mtot05", mtot05[10,10]
+print "icetot05", icetot05[10,10]
+
+## 4 --> an example of complete labelling
+## .... a rather unlikely example ....
+## .... but shows label ordering ....
+#########################################
 test = pp()
-test.file = [ \
-  "/planeto/jllmd/Earth/Nmix/F1525/output/diagfi4.nc", \
-  "/planeto/jllmd/Earth/Nmix/F1550/output/diagfi4.nc", \
-  "/planeto/jllmd/Earth/Nmix/F1500/output/diagfi4.nc" ]
-test.var = ["tsurf","ASR"]
-test.x = "-180.,175."
-test.y = "-90.,90."
-test.t = "0,1000"
-#test.verbose = True
-test.compute = "meanarea"
-test.get()
-
-# -----------------------------
-# self.allfield contains data !
-# -----------------------------
-# index ordering: 
-# - file
-# - var
-# - request t
-# - request z
-# - request y
-# - request x
-# - field dimensions
-# -----------------------------
-
-
-for ii in range(test.nfin):
-    print "tsurf",test.allfield[ii][0][0][0][0][0]
-    print "asr",test.allfield[ii][1][0][0][0][0]
-
-#mpl.plot(tsurf,asr)
-#mpl.show()
+test.file = ["/home/aymeric/Big_Data/DATAPLOT/diagfired.nc","/home/aymeric/Big_Data/DATAPLOT/diagfired.nc"]
+test.var = ["u","v"]
+test.x = [10.,20.]
+test.y = [10.,20.]
+test.z = [10.,20.]
+test.t = [0.4,0.5]
+print "... please wait. this one is a bit stupid..."
+allf,lab = test.getfl()
+## note label ordering: file -- var -- x -- y -- z -- t
+l1 = "_f=#2_v=u_x=10.0_y=10.0_z=20.0_t=0.4_"
+l2 = "_f=#2_v=v_x=10.0_y=10.0_z=20.0_t=0.4_"
+u_example = allf[lab.index(l1)]
+v_example = allf[lab.index(l2)]
+print l1, u_example
+print l2, v_example