Plot docs

doc/api/magic-plot.md

@@ -293,3 +293,35 @@ You can also show the percentage on top of the pie using the `S`/`show-numbers`
 ```{code-cell} ipython3
 %sqlplot pie --table penguins.csv --column species -S
 ```
+
+## Parametrizing arguments
+
+JupySQL supports variable expansion of arguments in the form of `{{variable}}`. This allows the user to specify arguments with placeholders that can be replaced by variables dynamically.
+
+Currently, `table`, `column`, `schema` and `with` can be parametrized. Let's see an example:
+
+```{code-cell} ipython3
+%%sql 
+DROP TABLE IF EXISTS penguins;
+CREATE SCHEMA IF NOT EXISTS s1;
+CREATE TABLE s1.penguins (
+    species VARCHAR(255),
+    island VARCHAR(255),
+    bill_length_mm DECIMAL(5, 2),
+    bill_depth_mm DECIMAL(5, 2),
+    flipper_length_mm DECIMAL(5, 2),
+    body_mass_g INTEGER,
+    sex VARCHAR(255)
+);
+COPY s1.penguins FROM 'penguins.csv' WITH (FORMAT CSV, HEADER TRUE);
+```
+
+```{code-cell} ipython3
+table = "penguins"
+schema = "s1"
+column = "bill_length_mm"
+```
+
+```{code-cell} ipython3
+%sqlplot boxplot --table {{penguins}} --schema {{schema}} --column {{column}}
+```

src/sql/magic_plot.py

@@ -86,7 +86,7 @@ def execute(self, line="", cell="", local_ns=None):
 
         cmd = SQLPlotCommand(self, line)
 
-        util.expand_args(cmd.args, user_ns, ["table", "column", "with_"])
+        util.expand_args(cmd.args, user_ns, ["table", "column", "schema", "with_"])
 
         if len(cmd.args.column) == 1:
             column = cmd.args.column[0]