def simple():import pandas as pdimport matplotlib.pyplot as pltimport pandas_aliveimport numpy as np# Physical constantsg = 9.81L = .4mu = 0.2THETA_0 = np.pi * 70 / 180 # init angle = 70degsTHETA_DOT_0 = 0 # no init angVelDELTA_T = 0.01 # time steppingT = 1.5 # time period# Definition of ODE (ordinary differential equation)def get_theta_double_dot(theta, theta_dot):return -mu * theta_dot - (g / L) * np.sin(theta)# Solution to the differential equationdef pendulum(t):# initialise changing valuestheta = THETA_0theta_dot = THETA_DOT_0delta_t = DELTA_Tang = []ang_vel = []ang_acc = []times = []for time in np.arange(0, t, delta_t):theta_double_dot = get_theta_double_dot(theta, theta_dot)theta += theta_dot * delta_ttheta_dot += theta_double_dot * delta_ttimes.append(time)ang.append(theta)ang_vel.append(theta_dot)ang_acc.append(theta_double_dot)data = np.array([ang, ang_vel, ang_acc])return pd.DataFrame(data=data.T, index=np.array(times), columns=["angle", "ang_vel", "ang_acc"])# units used for ref: ["angle [rad]", "ang_vel [rad/s]", "ang_acc [rad/s^2]"]df = pendulum(T)df.index.names = ["Time (s)"]print(df)# generate dataFrame for animated bubble plotdf2 = pd.DataFrame(index=df.index)df2["dx (m)"] = L * np.sin(df["angle"])df2["dy (m)"] = -L * np.cos(df["angle"])df2["ang_vel"] = abs(df["ang_vel"])df2["size"] = df2["ang_vel"] * 100 # scale angular vels to get nice size on bubble plotprint(df2)# static pandas plots## print(plt.style.available)# NOTE: 2 lines below required in Jupyter to switch styles correctlyplt.rcParams.update(plt.rcParamsDefault)plt.style.use("ggplot") # set plot stylefig, (ax1a, ax2b) = plt.subplots(1, 2, figsize=(8, 4), dpi=100) # 1 row, 2 subplots# fig.subplots_adjust(wspace=0.1) # space subplots in rowfig.set_tight_layout(True)fontsize = "small"df.plot(ax=ax1a).legend(fontsize=fontsize)ax1a.set_title("Outputs vs Time", fontsize="medium")ax1a.set_xlabel('Time [s]', fontsize=fontsize)ax1a.set_ylabel('Amplitudes', fontsize=fontsize);df.plot(ax=ax2b, x="angle", y=["ang_vel", "ang_acc"]).legend(fontsize=fontsize)ax2b.set_title("Outputs vs Angle | Phase-Space", fontsize="medium")ax2b.set_xlabel('Angle [rad]', fontsize=fontsize)ax2b.set_ylabel('Angular Velocity / Acc', fontsize=fontsize)# sample scatter plot with colorbarfig, ax = plt.subplots()sc = ax.scatter(df2["dx (m)"], df2["dy (m)"], s=df2["size"] * .1, c=df2["ang_vel"], cmap="jet")cbar = fig.colorbar(sc)cbar.set_label(label="ang_vel [rad/s]", fontsize="small")# sc.set_clim(350, 400)ax.tick_params(labelrotation=0, labelsize="medium")ax_scale = 1.ax.set_xlim(-L * ax_scale, L * ax_scale)ax.set_ylim(-L * ax_scale - 0.1, L * ax_scale - 0.1)# make axes square: a circle shows as a circleax.set_aspect(1 / ax.get_data_ratio())ax.arrow(0, 0, df2["dx (m)"].iloc[-1], df2["dy (m)"].iloc[-1],color="dimgray", ls=":", lw=2.5, width=.0, head_width=0, zorder=-1)ax.text(0, 0.15, s="size and colour of pendulum bob\nbased on pd column\nfor angular velocity",ha='center', va='center')# plt.show()dpi = 100ax_scale = 1.1figsize = (3, 3)fontsize = "small"# set up figure to pass onto `pandas_alive`# NOTE: by using Figure (capital F) instead of figure() `FuncAnimation` seems to run twice as fast!# fig1, ax1 = plt.subplots()fig1 = plt.Figure()ax1 = fig1.add_subplot()fig1.set_size_inches(figsize)ax1.set_title("Simple pendulum animation, L=" + str(L) + "m", fontsize="medium")ax1.set_xlabel("Time (s)", color='dimgray', fontsize=fontsize)ax1.set_ylabel("Amplitudes", color='dimgray', fontsize=fontsize)ax1.tick_params(labelsize=fontsize)# pandas_aliveline_chart = df.plot_animated(filename="pend-line.gif", kind='line', period_label={'x': 0.05, 'y': 0.9},steps_per_period=1, interpolate_period=False, period_length=50,period_fmt='Time:{x:10.2f}',enable_progress_bar=True, fixed_max=True, dpi=100, fig=fig1)plt.close()# Video('examples/pend-line.mp4', html_attributes="controls muted autoplay")# set up and generate animated scatter plot## set up figure to pass onto `pandas_alive`# NOTE: by using Figure (capital F) instead of figure() `FuncAnimation` seems to run twice as fast!fig1sc = plt.Figure()ax1sc = fig1sc.add_subplot()fig1sc.set_size_inches(figsize)ax1sc.set_title("Simple pendulum animation, L=" + str(L) + "m", fontsize="medium")ax1sc.set_xlabel("Time (s)", color='dimgray', fontsize=fontsize)ax1sc.set_ylabel("Amplitudes", color='dimgray', fontsize=fontsize)ax1sc.tick_params(labelsize=fontsize)# pandas_alivescatter_chart = df.plot_animated(filename="pend-scatter.gif", kind='scatter', period_label={'x': 0.05, 'y': 0.9},steps_per_period=1, interpolate_period=False, period_length=50,period_fmt='Time:{x:10.2f}',enable_progress_bar=True, fixed_max=True, dpi=100, fig=fig1sc, size="ang_vel")plt.close()print("Points size follows one of the pd columns: ang_vel")# Video('./pend-scatter.gif', html_attributes="controls muted autoplay")# set up and generate animated bar race chart## set up figure to pass onto `pandas_alive`# NOTE: by using Figure (capital F) instead of figure() `FuncAnimation` seems to run twice as fast!fig2 = plt.Figure()ax2 = fig2.add_subplot()fig2.set_size_inches(figsize)ax2.set_title("Simple pendulum animation, L=" + str(L) + "m", fontsize="medium")ax2.set_xlabel("Amplitudes", color='dimgray', fontsize=fontsize)ax2.set_ylabel("", color='dimgray', fontsize="x-small")ax2.tick_params(labelsize=fontsize)# pandas_aliverace_chart = df.plot_animated(filename="pend-race.gif", kind='race', period_label={'x': 0.05, 'y': 0.9},steps_per_period=1, interpolate_period=False, period_length=50,period_fmt='Time:{x:10.2f}',enable_progress_bar=True, fixed_max=False, dpi=100, fig=fig2)plt.close()# set up and generate bubble animated plot## set up figure to pass onto `pandas_alive`# NOTE: by using Figure (capital F) instead of figure() `FuncAnimation` seems to run twice as fast!fig3 = plt.Figure()ax3 = fig3.add_subplot()fig3.set_size_inches(figsize)ax3.set_title("Simple pendulum animation, L=" + str(L) + "m", fontsize="medium")ax3.set_xlabel("Hor Displacement (m)", color='dimgray', fontsize=fontsize)ax3.set_ylabel("Ver Displacement (m)", color='dimgray', fontsize=fontsize)# limits & ratio below get the graph squareax3.set_xlim(-L * ax_scale, L * ax_scale)ax3.set_ylim(-L * ax_scale - 0.1, L * ax_scale - 0.1)ratio = 1. # this is visual ratio of axesax3.set_aspect(ratio / ax3.get_data_ratio())ax3.arrow(0, 0, df2["dx (m)"].iloc[-1], df2["dy (m)"].iloc[-1],color="dimgray", ls=":", lw=1, width=.0, head_width=0, zorder=-1)# pandas_alivebubble_chart = df2.plot_animated(kind="bubble", filename="pend-bubble.gif",x_data_label="dx (m)", y_data_label="dy (m)",size_data_label="size", color_data_label="ang_vel", cmap="jet",period_label={'x': 0.05, 'y': 0.9}, vmin=None, vmax=None,steps_per_period=1, interpolate_period=False, period_length=50, period_fmt='Time:{x:10.2f}s',enable_progress_bar=True, fixed_max=False, dpi=dpi, fig=fig3)plt.close()print("Bubble size & colour animates with pd data column for ang_vel.")# Combined plots#fontsize = "x-small"# Otherwise titles overlap and subplots_adjust does nothingfrom matplotlib import rcParamsrcParams.update({"figure.autolayout": False})figs = plt.Figure(figsize=(9, 4), dpi=100)figs.subplots_adjust(wspace=0.1)gs = figs.add_gridspec(2, 2)ax1 = figs.add_subplot(gs[0, 0])ax1.set_xlabel("Time(s)", color='dimgray', fontsize=fontsize)ax1.set_ylabel("Amplitudes", color='dimgray', fontsize=fontsize)ax1.tick_params(labelsize=fontsize)ax2 = figs.add_subplot(gs[1, 0])ax2.set_xlabel("Amplitudes", color='dimgray', fontsize=fontsize)ax2.set_ylabel("", color='dimgray', fontsize=fontsize)ax2.tick_params(labelsize=fontsize)ax3 = figs.add_subplot(gs[:, 1])ax3.set_xlabel("Hor Displacement (m)", color='dimgray', fontsize=fontsize)ax3.set_ylabel("Ver Displacement (m)", color='dimgray', fontsize=fontsize)ax3.tick_params(labelsize=fontsize)# limits & ratio below get the graph squareax3.set_xlim(-L * ax_scale, L * ax_scale)ax3.set_ylim(-L * ax_scale - 0.1, L * ax_scale - 0.1)ratio = 1. # this is visual ratio of axesax3.set_aspect(ratio / ax3.get_data_ratio())line_chart.ax = ax1race_chart.ax = ax2bubble_chart.ax = ax3plots = [line_chart, race_chart, bubble_chart]# pandas_alive combined using custom figurepandas_alive.animate_multiple_plots(filename='pend-combined.gif', plots=plots, custom_fig=figs, dpi=100, enable_progress_bar=True,adjust_subplot_left=0.2, adjust_subplot_right=None,title="Simple pendulum animations, L=" + str(L) + "m", title_fontsize="medium")plt.close()
评论
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
