update plotting grouped data

UtrechtUniversity · Feb 22, 2024 · 0c43d53 · 0c43d53
1 parent 02b9ecd
commit 0c43d53
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/book/data-science-with-pandas-4.ipynb b/book/data-science-with-pandas-4.ipynb
@@ -470,8 +470,8 @@
    "source": [
     "fig, ax = plt.subplots()\n",
     "\n",
-    "for i, group in list(surveys.groupby('sex')):\n",
-    "    ax.scatter(group['hindfoot_length'], group['weight'], alpha=0.5, label = group['sex'].iloc[0])\n",
+    "for item, group in list(surveys.groupby('sex')):\n",
+    "    ax.scatter(group['hindfoot_length'], group['weight'], alpha=0.5, label = item)\n",
     "    \n",
     "ax.legend(title='sex')\n",
     "ax.grid()\n",