From 3c3e804251883b41553fcac8131efcd916fee5fb Mon Sep 17 00:00:00 2001 From: scuti Date: Sun, 27 Apr 2025 12:42:39 -0700 Subject: [PATCH] Simplified code. Removed to_frame(). --- stackoverflow-survey.ipynb | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/stackoverflow-survey.ipynb b/stackoverflow-survey.ipynb index dadb6fc..0277e01 100644 --- a/stackoverflow-survey.ipynb +++ b/stackoverflow-survey.ipynb @@ -221,8 +221,8 @@ " random=333, risky=0,\n", " color='red', name='Regression Line' ):\n", " df = self.devs # .sort_values(by = self.key2)\n", - " X = df[self.key_x].to_frame()\n", - " y = df[self.key_y].to_frame()\n", + " X = df[[self.key_x]]\n", + " y = df[[self.key_y]]\n", "\n", " # not recommended\n", " # carries risk of model training on sorted order\n", @@ -270,8 +270,8 @@ "\n", " def run_log_regression(self, color='pink', nodraw=True):\n", " df = self.devs\n", - " X = df[self.key_x].to_frame().sort_values(by=self.key_x)\n", - " y = df[self.key_y].to_frame().sort_values(by=self.key_y)\n", + " X = df[[self.key_x]].sort_values(by=self.key_x)\n", + " y = df[[self.key_y]].sort_values(by=self.key_y)\n", "\n", " X_log = np.log(X)\n", " x_fit = np.linspace(1, 40, len(y)).reshape(-1, 1)\n",