Changeset 563 for branches

Timestamp:

09/18/12 15:04:55 (4 years ago)

Author:

mmckerns

Message:

enable selection of color for datapoints and cone

File:

• branches/UQ/math/legacy/plot_data.py (modified) (6 diffs)

branches/UQ/math/legacy/plot_data.py

@@ -67 +67 @@
 
 #### plotting the cones ####
-def plot_cones(ax, data, slope, bounds):
+def plot_cones(ax, data, slope, bounds, color='0.75'):
   """plot double cones for a given dataset
 
@@ -74 +74 @@
   slope -- slope multiplier for cone on the X,Y,Z axes (for mesh construction)
   bounds -- list of tuples of bounds for the plot; (lower,upper) for each axis
+  color -- string name (or rbg value) of color to use for datapoints
 """
   cone = cone_builder(slope, bounds)
@@ -81 +82 @@
     if _cone:
       X,Z,Y = _cone
-      ax.plot_surface(X, Z,Y, rstride=1, cstride=1, color="0.75") 
+      ax.plot_surface(X, Z,Y, rstride=1, cstride=1, color=color) 
     _cone = cone(datapt, top=False)
     if _cone:
       X,Z,Y = _cone
-      ax.plot_surface(X, Z,Y, rstride=1, cstride=1, color="0.75")
+      ax.plot_surface(X, Z,Y, rstride=1, cstride=1, color=color)
   return ax
 
-def plot_data(ax, data, bounds):
+def plot_data(ax, data, bounds, color='red'):
   """plot datapoints for a given dataset
 
@@ -94 +95 @@
   data -- list of datapoints, where datapoints are 3-tuples (i.e. x,y,z)
   bounds -- list of tuples of bounds for the plot; (lower,upper) for each axis
+  color -- string name (or rbg value) of color to use for datapoints
 """
   strict = True # always respect the bounds
@@ -107 +109 @@
     else:
       ax.plot([datapt[0]], [datapt[2]], [datapt[1]], \
-              color='r', marker='o', markersize=10)
+              color=color, marker='o', markersize=10)
   return ax
 
@@ -184 +186 @@
     plot_cones(ax, coords, slope, bounds)
   plot_data(ax, coords, bounds)
-  clip_axes(ax, bounds)
+ #clip_axes(ax, bounds)
   plt.show()