relate window_length to the input data in savgol_filter

ssciwr · Jul 9, 2024 · b130f68 · b130f68
1 parent 122197b
commit b130f68
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/docs/notebooks/path_and_velocity.ipynb b/docs/notebooks/path_and_velocity.ipynb
@@ -365,7 +365,7 @@
     "            timestamps = concatenate(timestamps, group.to_center_timestamps.iloc[i])\n",
     "        velocity = get_velocity(timestamps, positions)\n",
     "        ax.plot(timestamps[:-1], velocity, color=color)\n",
-    "        positions[:, 1] = savgol_filter(positions[:, 1], len(positions[:, 0]), 8)\n",
+    "        positions[:, 1] = savgol_filter(positions[:, 1], len(positions[:, 1]), 8)\n",
     "        velocity_with_filtered_positions = get_velocity(timestamps, positions)\n",
     "        ax.plot(\n",
     "            timestamps[:-1],\n",
@@ -417,9 +417,7 @@
     "            positions = concatenate(positions, group.to_center_mouse_positions.iloc[i])\n",
     "            timestamps = concatenate(timestamps, group.to_center_timestamps.iloc[i])\n",
     "        velocity = get_velocity(timestamps, positions)\n",
-    "        filtered_velocity = savgol_filter(\n",
-    "            velocity, len(timestamps[:-1]), len(timestamps[:-1]) - 1\n",
-    "        )\n",
+    "        filtered_velocity = savgol_filter(velocity, len(velocity), len(velocity) - 1)\n",
     "        ax.plot(timestamps[:-1], filtered_velocity, color=\"black\", linestyle=\"dashed\")\n",
     "        ax.plot(timestamps[:-1], velocity, color=color)\n",
     "\n",
@@ -431,6 +429,13 @@
     "warnings.filterwarnings(\"ignore\")\n",
     "plt.show()"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {