added lines_bars_and_markers/errorbar_subsample

Author: MamoruSobue

gallery/lines_bars_and_markers/CMakeLists.txt

@@ -7,9 +7,10 @@ add_demo(fill_betweenx_demo fill_betweenx_demo.cpp)
 add_demo(scatter_with_legend scatter_with_legend.cpp)
 add_demo(scatter_hist scatter_hist.cpp)
 add_demo(errorbar_limits_simple errorbar_limits_simple.cpp)
+add_demo(errorbar_subsample errorbar_subsample.cpp)
 
 add_custom_target(lines_bars_and_markers
-  DEPENDS bar_label_demo fill simple_plot scatter_symbol fill_between_demo fill_betweenx_demo scatter_with_legend scatter_hist errorbar_limits_simple
+  DEPENDS bar_label_demo fill simple_plot scatter_symbol fill_between_demo fill_betweenx_demo scatter_with_legend scatter_hist errorbar_limits_simple errorbar_subsample
   COMMAND bar_label_demo
   COMMAND fill
   COMMAND simple_plot
@@ -19,6 +20,7 @@ add_custom_target(lines_bars_and_markers
   COMMAND scatter_with_legend
   COMMAND scatter_hist
   COMMAND errorbar_limits_simple
+  COMMAND errorbar_subsample
   COMMENT "running lines_bars_and_markers"
   WORKING_DIRECTORY "${CMAKE_CURRENT_SOURCE_DIR}/../images"
 )

gallery/lines_bars_and_markers/errorbar_subsample.cpp

@@ -0,0 +1,54 @@
+// example from
+// https://matplotlib.org/stable/gallery/lines_bars_and_markers/errorbar_subsample.html
+
+#include <pybind11/embed.h>
+#include <pybind11/stl.h>
+
+#include <matplotlibcpp17/pyplot.h>
+
+#include <xtensor/xbuilder.hpp>
+#include <xtensor/xmath.hpp>
+
+#include <vector>
+
+namespace py = pybind11;
+using namespace py::literals;
+using namespace std;
+using namespace matplotlibcpp17;
+
+int main() {
+  auto x_ = xt::arange(0.1, 4.0, 0.1);
+  auto y1_ = xt::exp(-1.0 * x_);
+  auto y2_ = xt::exp(-0.5 * x_);
+  auto y1err_ = 0.1 + 0.1 * xt::sqrt(x_);
+  auto y2err_ = 0.1 + 0.1 * xt::sqrt(x_ / 2.0);
+  vector<double> x(x_.begin(), x_.end()), y1(y1_.begin(), y1_.end()),
+      y2(y2_.begin(), y2_.end()), y1err(y1err_.begin(), y1err_.end()),
+      y2err(y2err_.begin(), y2err_.end());
+
+  py::scoped_interpreter guard{};
+  auto plt = pyplot::import();
+  auto [fig, axs] = plt.subplots(
+      1, 3, Kwargs("sharex"_a = true, "figsize"_a = py::make_tuple(12, 6)));
+  auto ax0 = axs[0], ax1 = axs[1], ax2 = axs[2];
+  ax0.set_title(Args("all errorbars"));
+  ax0.errorbar(Args(x, y1), Kwargs("yerr"_a = y1err));
+  ax0.errorbar(Args(x, y1), Kwargs("yerr"_a = y2err));
+
+  ax1.set_title(Args("only every 6th errorbar"));
+  ax1.errorbar(Args(x, y1), Kwargs("yerr"_a = y1err, "errorevery"_a = 6));
+  ax1.errorbar(Args(x, y2), Kwargs("yerr"_a = y2err, "errorevery"_a = 6));
+
+  ax2.set_title(Args("second seris shifted by 3"));
+  ax2.errorbar(Args(x, y1),
+               Kwargs("yerr"_a = y1err, "errorevery"_a = py::make_tuple(0, 6)));
+  ax2.errorbar(Args(x, y2),
+               Kwargs("yerr"_a = y2err, "errorevery"_a = py::make_tuple(3, 6)));
+
+  fig.suptitle(Args("Errorbar subsampling"));
+#if USE_GUI
+  plt.show();
+#else
+  plt.savefig(Args("erorbar_subsample.png"));
+#endif
+}

include/matplotlibcpp17/axes.h

@@ -69,6 +69,10 @@ struct DECL_STRUCT_ATTR Axes : public BaseWrapper {
   pybind11::object contour(const pybind11::tuple &args = pybind11::tuple(),
                            const pybind11::dict &kwargs = pybind11::dict());
 
+  // errorbar
+  pybind11::object errorbar(const pybind11::tuple &args = pybind11::tuple(),
+                            const pybind11::dict &kwargs = pybind11::dict());
+
   // fill
   pybind11::object fill(const pybind11::tuple &args = pybind11::tuple(),
                         const pybind11::dict &kwargs = pybind11::dict());
@@ -213,6 +217,7 @@ struct DECL_STRUCT_ATTR Axes : public BaseWrapper {
 #endif
     LOAD_FUNC_ATTR(barh, self);
     LOAD_FUNC_ATTR(contour, self);
+    LOAD_FUNC_ATTR(errorbar, self);
     LOAD_FUNC_ATTR(fill, self);
     LOAD_FUNC_ATTR(fill_between, self);
     LOAD_FUNC_ATTR(fill_betweenx, self);
@@ -258,6 +263,7 @@ struct DECL_STRUCT_ATTR Axes : public BaseWrapper {
   pybind11::object bar_label_attr;
   pybind11::object barh_attr;
   pybind11::object contour_attr;
+  pybind11::object errorbar_attr;
   pybind11::object fill_attr;
   pybind11::object fill_between_attr;
   pybind11::object fill_betweenx_attr;
@@ -353,6 +359,13 @@ pybind11::object Axes::contour(const pybind11::tuple &args,
   return obj;
 }
 
+// errorbar
+pybind11::object Axes::errorbar(const pybind11::tuple &args,
+                                const pybind11::dict &kwargs) {
+  pybind11::object obj = errorbar_attr(*args, **kwargs);
+  return obj;
+}
+
 // fill
 pybind11::object Axes::fill(const pybind11::tuple &args,
                             const pybind11::dict &kwargs) {