Pandas 2.0 compat in docs

mwaskom · Apr 3, 2023 · 4f72554 · 4f72554
1 parent 011caac
commit 4f72554
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/doc/_docstrings/relplot.ipynb b/doc/_docstrings/relplot.ipynb
@@ -218,7 +218,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "flights_wide = sns.load_dataset(\"flights\").pivot(\"year\", \"month\", \"passengers\")"
+    "flights_wide = (\n",
+    "    sns.load_dataset(\"flights\")\n",
+    "    .pivot(index=\"year\", columns=\"month\", values=\"passengers\")\n",
+    ")"
    ]
   },
   {

diff --git a/doc/_tutorial/data_structure.ipynb b/doc/_tutorial/data_structure.ipynb
@@ -344,7 +344,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "flights_avg = flights.groupby(\"year\").mean()\n",
+    "flights_avg = flights.groupby(\"year\").mean(numeric_only=True)\n",
     "sns.relplot(data=flights_avg, x=\"year\", y=\"passengers\", kind=\"line\")"
    ]
   },

diff --git a/examples/spreadsheet_heatmap.py b/examples/spreadsheet_heatmap.py
@@ -9,7 +9,10 @@
 
 # Load the example flights dataset and convert to long-form
 flights_long = sns.load_dataset("flights")
-flights = flights_long.pivot("month", "year", "passengers")
+flights = (
+    flights_long
+    .pivot(index="month", columns="year", values="passengers")
+)
 
 # Draw a heatmap with the numeric values in each cell
 f, ax = plt.subplots(figsize=(9, 6))