\n",
" \n",
" 
\n",
"
\n",
" "
],
"text/plain": [
"Canvas(toolbar=Toolbar(toolitems=[('Home', 'Reset original view', 'home', 'home'), ('Back', 'Back to previous …"
]
},
"metadata": {},
"output_type": "display_data"
}
],
"source": [
"x_data = []\n",
"y_data = []\n",
"\n",
"fig, ax = plt.subplots() # A tuple unpacking to unpack the only fig\n",
"ax.set_xlim(30, 250)\n",
"ax.set_ylim(5, 50)\n",
"# Plotting \n",
"scatter, = ax.plot([], [], 'go', label='Horsepower vs. Miles_per_Gallon')\n",
"line, = ax.plot([], [], 'r', label='Linear Regression')\n",
"ax.legend()\n",
"\n",
"reg = LinearRegression()\n",
"\n",
"def animate(frame_num):\n",
" # Adding data\n",
" x_data.append(x[frame_num])\n",
" y_data.append(y[frame_num])\n",
" # Convert data to numpy array\n",
" x_train = np.array(x_data).reshape(-1, 1)\n",
" y_train = np.array(y_data).reshape(-1, 1)\n",
" # Fit values to a linear regression\n",
" reg.fit(x_train, y_train)\n",
"\n",
" # update data for scatter plot\n",
" scatter.set_data((x_data, y_data))\n",
" # Predict value and update data for line plot\n",
" line.set_data((list(range(250)), reg.predict(np.array([entry for entry in range(250)]).reshape(-1, 1))))\n",
"\n",
"anim = FuncAnimation(fig, animate, frames=len(x), interval=20)\n",
"plt.show()"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"## 3. Animations with embedded HTML5 video"
]
},
{
"cell_type": "code",
"execution_count": 10,
"metadata": {},
"outputs": [],
"source": [
"import matplotlib.pyplot as plt\n",
"from matplotlib.animation import FuncAnimation\n",
"from IPython import display\n",
"\n",
"# Turn off matplotlib plot in Notebook\n",
"plt.ioff()\n",
"# Pass the ffmpeg path\n",
"plt.rcParams['animation.ffmpeg_path'] = '/usr/local/bin/ffmpeg'"
]
},
{
"cell_type": "code",
"execution_count": 11,
"metadata": {},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"/opt/conda/lib/python3.10/site-packages/matplotlib/animation.py:879: UserWarning: Animation was deleted without rendering anything. This is most likely not intended. To prevent deletion, assign the Animation to a variable, e.g. `anim`, that exists until you output the Animation using `plt.show()` or `anim.save()`.\n",
" warnings.warn(\n"
]
},
{
"data": {
"text/html": [
""
],
"text/plain": [
"\n",
"