zebra/pyzebra
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add Legend to main plots

Browse Source
This commit is contained in:
usov_i 2021-02-26 13:05:57 +01:00
parent 5dbb208e92
commit 966d6349df
2 changed files with 29 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
pyzebra/app/panel_ccl_integrate.py
View File

@ -19,6 +19,7 @@ from bokeh.models import (
Dropdown, Dropdown,
FileInput, FileInput,
Grid, Grid,
Legend,
Line, Line,
LinearAxis, LinearAxis,
MultiSelect, MultiSelect,
@ -226,14 +227,18 @@ def create():
plot.add_layout(Grid(dimension=1, ticker=BasicTicker())) plot.add_layout(Grid(dimension=1, ticker=BasicTicker()))
plot_scatter_source = ColumnDataSource(dict(x=[0], y=[0], y_upper=[0], y_lower=[0])) plot_scatter_source = ColumnDataSource(dict(x=[0], y=[0], y_upper=[0], y_lower=[0]))
plot.add_glyph(plot_scatter_source, Scatter(x="x", y="y", line_color="steelblue")) plot_scatter = plot.add_glyph(
plot_scatter_source, Scatter(x="x", y="y", line_color="steelblue")
)
plot.add_layout(Whisker(source=plot_scatter_source, base="x", upper="y_upper", lower="y_lower")) plot.add_layout(Whisker(source=plot_scatter_source, base="x", upper="y_upper", lower="y_lower"))
plot_fit_source = ColumnDataSource(dict(x=[0], y=[0])) plot_fit_source = ColumnDataSource(dict(x=[0], y=[0]))
plot.add_glyph(plot_fit_source, Line(x="x", y="y")) plot_fit = plot.add_glyph(plot_fit_source, Line(x="x", y="y"))
plot_bkg_source = ColumnDataSource(dict(x=[0], y=[0])) plot_bkg_source = ColumnDataSource(dict(x=[0], y=[0]))
plot.add_glyph(plot_bkg_source, Line(x="x", y="y", line_color="green", line_dash="dashed")) plot_bkg = plot.add_glyph(
plot_bkg_source, Line(x="x", y="y", line_color="green", line_dash="dashed")
)
numfit_min_span = Span(location=None, dimension="height", line_dash="dashed") numfit_min_span = Span(location=None, dimension="height", line_dash="dashed")
plot.add_layout(numfit_min_span) plot.add_layout(numfit_min_span)
@ -241,6 +246,13 @@ def create():
numfit_max_span = Span(location=None, dimension="height", line_dash="dashed") numfit_max_span = Span(location=None, dimension="height", line_dash="dashed")
plot.add_layout(numfit_max_span) plot.add_layout(numfit_max_span)
plot.add_layout(
Legend(
items=[("data", [plot_scatter]), ("best fit", [plot_fit]), ("background", [plot_bkg])],
location="top_left",
)
)
plot.add_tools(PanTool(), WheelZoomTool(), ResetTool()) plot.add_tools(PanTool(), WheelZoomTool(), ResetTool())
plot.toolbar.logo = None plot.toolbar.logo = None

18
pyzebra/app/panel_param_study.py
View File

@ -20,6 +20,7 @@ from bokeh.models import (
Dropdown, Dropdown,
FileInput, FileInput,
Grid, Grid,
Legend,
Line, Line,
LinearAxis, LinearAxis,
MultiLine, MultiLine,
@ -270,15 +271,17 @@ def create():
plot.add_layout(Grid(dimension=1, ticker=BasicTicker())) plot.add_layout(Grid(dimension=1, ticker=BasicTicker()))
plot_scatter_source = ColumnDataSource(dict(x=[0], y=[0], y_upper=[0], y_lower=[0])) plot_scatter_source = ColumnDataSource(dict(x=[0], y=[0], y_upper=[0], y_lower=[0]))
plot.add_glyph(plot_scatter_source, Scatter(x="x", y="y", line_color="steelblue")) plot_scatter = plot.add_glyph(
plot_scatter_source, Scatter(x="x", y="y", line_color="steelblue")
)
plot.add_layout(Whisker(source=plot_scatter_source, base="x", upper="y_upper", lower="y_lower")) plot.add_layout(Whisker(source=plot_scatter_source, base="x", upper="y_upper", lower="y_lower"))
plot_fit_source = ColumnDataSource(dict(x=[0], y=[0])) plot_fit_source = ColumnDataSource(dict(x=[0], y=[0]))
plot.add_glyph(plot_fit_source, Line(x="x", y="y")) plot_fit = plot.add_glyph(plot_fit_source, Line(x="x", y="y"))
plot_bkg_source = ColumnDataSource(dict(x=[0], y=[0])) plot_bkg_source = ColumnDataSource(dict(x=[0], y=[0]))
plot.add_glyph( plot_bkg = plot.add_glyph(
plot_bkg_source, Line(x="x", y="y", line_color="green", line_dash="dashed"), plot_bkg_source, Line(x="x", y="y", line_color="green", line_dash="dashed")
) )
numfit_min_span = Span(location=None, dimension="height", line_dash="dashed") numfit_min_span = Span(location=None, dimension="height", line_dash="dashed")
@ -287,6 +290,13 @@ def create():
numfit_max_span = Span(location=None, dimension="height", line_dash="dashed") numfit_max_span = Span(location=None, dimension="height", line_dash="dashed")
plot.add_layout(numfit_max_span) plot.add_layout(numfit_max_span)
plot.add_layout(
Legend(
items=[("data", [plot_scatter]), ("best fit", [plot_fit]), ("background", [plot_bkg])],
location="top_left",
)
)
plot.add_tools(PanTool(), WheelZoomTool(), ResetTool()) plot.add_tools(PanTool(), WheelZoomTool(), ResetTool())
plot.toolbar.logo = None plot.toolbar.logo = None