%sqlplot bar and pie charts

.gitignore

@@ -62,8 +62,9 @@ nosetests.xml
 
 .venv
 
-# Do not include test output
-/result_images/
+
+# Do not include test output of matplotlib
+result_images
 
 # Ignore Github codespace build artifact
-oryx-build-commands.txt
+oryx-build-commands.txt

CHANGELOG.md

@@ -16,8 +16,8 @@
 * [Feature] Adds `--persist-replace` argument to replace existing tables when persisting data frames ([#440](https://github.com/ploomber/jupysql/issues/440))
 * [Fix] Fix error when checking if custom connection was PEP 249 Compliant ([#517](https://github.com/ploomber/jupysql/issues/517))
 * [Doc] documenting how to manage connections with `Connection` object ([#282](https://github.com/ploomber/jupysql/issues/282))
-
 * [Feature] Github Codespace (Devcontainer) support for development (by [@jorisroovers](https://github.com/jorisroovers)) ([#484](https://github.com/ploomber/jupysql/issues/484))
+* [Feature] Added bar plot and pie charts to %sqlplot ([#417](https://github.com/ploomber/jupysql/issues/417))
 
 ## 0.7.5 (2023-05-24)
 

doc/api/magic-plot.md

@@ -160,3 +160,92 @@ ax = %sqlplot histogram --table no_nulls --column body_mass_g --with no_nulls
 ax.set_title("Body mass (grams)")
 _ = ax.grid()
 ```
+## `%sqlplot bar`
+
+```{versionadded} 0.7.6
+```
+
+Shortcut: `%sqlplot bar`
+
+`-t`/`--table` Table to use (if using DuckDB: path to the file to query)
+
+`-c`/`--column` Column to plot.
+
+`-o`/`--orient` Barplot orientation (`h` for horizontal, `v` for vertical)
+
+`-w`/`--with` Use a previously saved query as input data
+
+`-S`/`--show-numbers` Show numbers on top of the bar
+
+Bar plot does not support NULL values, so we automatically remove them, when plotting.
+
+```{code-cell} ipython3
+%sqlplot bar --table penguins.csv --column species 
+```
+
+You can additionally pass two columns to bar plot i.e. `x` and `height` columns.
+
+```{code-cell} ipython3
+%%sql --save add_col --no-execute
+SELECT species, count(species) as cnt
+FROM penguins.csv
+group by species
+```
+
+```{code-cell} ipython3
+%sqlplot bar --table add_col --column species cnt --with add_col
+```
+
+You can also pass the orientation using the `orient` argument.
+
+```{code-cell} ipython3
+%sqlplot bar --table add_col --column species cnt --with add_col --orient h
+``` 
+
+You can also show the number on top of the bar using the `S`/`show-numbers` argument.
+
+```{code-cell} ipython3
+%sqlplot bar --table penguins.csv --column species -S
+```
+
+## `%sqlplot pie`
+
+```{versionadded} 0.7.6
+```
+
+Shortcut: `%sqlplot pie`
+
+`-t`/`--table` Table to use (if using DuckDB: path to the file to query)
+
+`-c`/`--column` Column to plot
+
+`-w`/`--with` Use a previously saved query as input data
+
+`-S`/`--show-numbers` Show the percentage on top of the pie
+
+Pie chart does not support NULL values, so we automatically remove them, when plotting the pie chart.
+
+```{code-cell} ipython3
+%sqlplot pie --table penguins.csv --column species
+```
+
+You can additionally pass two columns to bar plot i.e. `labels` and `x` columns.
+
+```{code-cell} ipython3
+%%sql --save add_col --no-execute
+SELECT species, count(species) as cnt
+FROM penguins.csv
+group by species
+```
+
+```{code-cell} ipython3
+%sqlplot pie --table add_col --column species cnt --with add_col
+```
+Here, `species` is the `labels` column and `cnt` is the `x` column.
+
+
+You can also show the percentage on top of the pie using the `S`/`show-numbers` argument.
+
+```{code-cell} ipython3
+%sqlplot pie --table penguins.csv --column species -S
+```

src/sql/magic_plot.py

@@ -17,6 +17,8 @@
 from sql import exceptions
 from sql import util
 
+SUPPORTED_PLOTS = ["histogram", "boxplot", "bar", "pie"]
+
 
 @magics_class
 class SqlPlotMagic(Magics, Configurable):
@@ -51,6 +53,12 @@ class SqlPlotMagic(Magics, Configurable):
         action="append",
         dest="with_",
     )
+    @argument(
+        "-S",
+        "--show-numbers",
+        action="store_true",
+        help="Show number of observations",
+    )
     @modify_exceptions
     def execute(self, line="", cell="", local_ns=None):
         """
@@ -65,8 +73,10 @@ def execute(self, line="", cell="", local_ns=None):
             column = cmd.args.column
 
         if not cmd.args.line:
+            plot_str = util.pretty_print(SUPPORTED_PLOTS, last_delimiter="or")
             raise exceptions.UsageError(
-                "Missing the first argument, must be: 'histogram' or 'boxplot'. "
+                "Missing the first argument, must be any of: "
+                f"{plot_str}\n"
                 "Example: %sqlplot histogram"
             )
 
@@ -92,7 +102,29 @@ def execute(self, line="", cell="", local_ns=None):
                 with_=cmd.args.with_,
                 conn=None,
             )
+        elif cmd.args.line[0] in {"bar"}:
+            util.is_table_exists(table, with_=cmd.args.with_)
+
+            return plot.bar(
+                table=table,
+                column=column,
+                with_=cmd.args.with_,
+                orient=cmd.args.orient,
+                show_num=cmd.args.show_numbers,
+                conn=None,
+            )
+        elif cmd.args.line[0] in {"pie"}:
+            util.is_table_exists(table, with_=cmd.args.with_)
+
+            return plot.pie(
+                table=table,
+                column=column,
+                with_=cmd.args.with_,
+                show_num=cmd.args.show_numbers,
+                conn=None,
+            )
         else:
+            plot_str = util.pretty_print(SUPPORTED_PLOTS, last_delimiter="or")
             raise exceptions.UsageError(
-                f"Unknown plot {cmd.args.line[0]!r}. Must be: 'histogram' or 'boxplot'"
+                f"Unknown plot {cmd.args.line[0]!r}. Must be any of: " f"{plot_str}"
             )