errorbar(), tests

Author: aplavin

ext/AxisKeysExt.jl

@@ -1,8 +1,8 @@
 module AxisKeysExt
-using AxisKeys: KeyedArray, axiskeys, dimnames
+using AxisKeys: KeyedArray, axiskeys, dimnames, keyless_unname
 using PyPlotUtils.IntervalSets
 using PyPlotUtils
-import PyPlotUtils: extent_ax, extent_arr, plot_ax, fill_between_ax, pcolormesh_ax
+import PyPlotUtils: extent_ax, extent_arr, plot_ax, fill_between_ax, errorbar_ax, pcolormesh_ax
 
 extent_arr(A::KeyedArray) = (extent_ax(axiskeys(A, 1))..., extent_ax(axiskeys(A, 2))...)
 
@@ -14,6 +14,10 @@ function fill_between_ax(A::KeyedArray; kwargs...)
     plt.fill_between(only(axiskeys(A)), leftendpoint.(A), rightendpoint.(A); kwargs...)
 end
 
+function errorbar_ax(A::KeyedArray; kwargs...)
+    errorbar(only(axiskeys(A)), keyless_unname(A); kwargs...)
+end
+
 function pcolormesh_ax(A::KeyedArray; kwargs...)
     res = plt.pcolormesh(
         axiskeys(A, 1),

src/PyPlotUtils.jl

@@ -23,7 +23,7 @@ export
     keep_plt_lims, set_xylims, lim_intersect, lim_union,
     set_xylabels, xylabels, label_log_scales, draw_text_along, legend_inline_right,
     imshow_ax, SymLog, ColorBar,
-    pcolormesh_ax, plot_ax, fill_between_ax,
+    pcolormesh_ax, plot_ax, fill_between_ax, errorbar, errorbar_ax,
     mpl_color, adjust_lightness,
     axplotfunc, add_zoom_patch,
     ScalebarArtist
@@ -202,8 +202,22 @@ end
 # XXX: generalize somehow? like axiskeys(pcolormesh)(A) or ...?
 function plot_ax end
 function fill_between_ax end
+function errorbar_ax end
 function pcolormesh_ax end
 
+errorbar(x, y; kwargs...) = errorbar([x], [y]; kwargs...)
+function errorbar(x::AbstractVector, y::AbstractVector; midfunc=mean, kwargs...)
+    xmids = midfunc.(x)
+    xerr = map(x) do x
+        abs.(extrema(x) .- midfunc(x))
+    end |> stack
+    ymids = midfunc.(y)
+    yerr = map(y) do x
+        abs.(extrema(x) .- midfunc(x))
+    end |> stack
+    plt.errorbar(xmids, ymids; xerr, yerr, kwargs...)
+end
+
 """    axplotfunc(f; ax=plt.gca(), n=10, [plot() kwargs...])
 Plot `y = f(x)` within the current axis limits. Uses `n` points that uniformly split the whole `x` interval. """
 function axplotfunc(f; ax=plt.gca(), n=10, kwargs...)

test/Project.toml

@@ -3,6 +3,7 @@ Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
 AxisKeys = "94b1ba4f-4ee9-5380-92f1-94cde586c3c5"
 Colors = "5ae59095-9a9b-59fe-a467-6f913c188581"
 CompatHelperLocal = "5224ae11-6099-4aaa-941d-3aab004bd678"
+IntervalSets = "8197267c-284f-5f27-9208-e0e47529a953"
 OffsetArrays = "6fe1bfb0-de20-5000-8ca7-80f57d26f881"
 PyPlot = "d330b81b-6aea-500a-939a-2ce795aea3ee"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
@@ -13,6 +14,7 @@ Aqua = "0.6"
 AxisKeys = "0.2"
 Colors = "0.12"
 CompatHelperLocal = "0.1"
+IntervalSets = "0.7"
 OffsetArrays = "1.10"
 PyPlot = "2.9"
 Unitful = "1.9"

test/runtests.jl

@@ -1,4 +1,5 @@
 using PyPlotUtils
+using IntervalSets
 using AxisKeys
 using OffsetArrays
 using Unitful
@@ -42,6 +43,15 @@ imshow_ax(KeyedArray([1 2; 3 4], a=(-5:-4)u"m", b=(0:1)u"m"))
 imshow_ax(KeyedArray([1 2; 3 4], a=(-5:-4)u"m", b=(0:1)u"m"), ColorBar())
 imshow_ax(KeyedArray([1 2; 3 4], a=(-5:-4)u"m", b=(0:1)u"m"), ColorBar(); norm=SymLog())
 
+plot_ax(KeyedArray([1, 2, 3], a=5:7))
+fill_between_ax(KeyedArray([1..2, 2..3, 3..4], a=5:7))
+errorbar_ax(KeyedArray([1..2, 2..3, 3..4], a=5:7))
+pcolormesh_ax(KeyedArray([1 2; 3 4], a=-5:-4, b=0:1))
+errorbar(1, 1..2)
+errorbar(1..2, 1)
+errorbar(1..2, 1..2)
+errorbar([1..2], [1..2])
+
 plt.gca().add_artist(ScalebarArtist([(1, "px")]))
 
 axplotfunc(sin)