Commit 167933fb authored by Sebastian Heimann's avatar Sebastian Heimann
Browse files

jointpar figures: add kwarg to include only certain parameters

parent 55842e44
...@@ -316,10 +316,11 @@ def draw_sequence_figures(model, plt, misfit_cutoff=None): ...@@ -316,10 +316,11 @@ def draw_sequence_figures(model, plt, misfit_cutoff=None):
def draw_jointpar_figures( def draw_jointpar_figures(
model, plt, misfit_cutoff=None, ibootstrap=None, color=None, model, plt, misfit_cutoff=None, ibootstrap=None, color=None,
exclude=None): exclude=None, include=None):
color = 'magnitude' color = 'misfit'
# exclude = ['duration'] # exclude = ['duration']
# include = ['magnitude', 'rel_moment_iso', 'rel_moment_clvd', 'depth']
neach = 6 neach = 6
figsize = (8, 8) figsize = (8, 8)
# cmap = cm.YlOrRd # cmap = cm.YlOrRd
...@@ -379,7 +380,7 @@ def draw_jointpar_figures( ...@@ -379,7 +380,7 @@ def draw_jointpar_figures(
iselected = 0 iselected = 0
for ipar in xrange(problem.ncombined): for ipar in xrange(problem.ncombined):
par = problem.combined[ipar] par = problem.combined[ipar]
if exclude and par.name in exclude: if exclude and par.name in exclude or include and par.name not in include:
continue continue
smap[iselected] = ipar smap[iselected] = ipar
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment