mirror of
https://github.com/gradio-app/gradio.git
synced 2024-12-09 02:00:44 +08:00
db54b7b76a
* Use lineplot for kinematics * Changelog
44 lines
1.0 KiB
Python
44 lines
1.0 KiB
Python
import pandas as pd
|
|
import numpy as np
|
|
|
|
import gradio as gr
|
|
|
|
|
|
def plot(v, a):
|
|
g = 9.81
|
|
theta = a / 180 * 3.14
|
|
tmax = ((2 * v) * np.sin(theta)) / g
|
|
timemat = tmax * np.linspace(0, 1, 40)
|
|
|
|
x = (v * timemat) * np.cos(theta)
|
|
y = ((v * timemat) * np.sin(theta)) - ((0.5 * g) * (timemat**2))
|
|
df = pd.DataFrame({"x": x, "y": y})
|
|
return df
|
|
|
|
|
|
demo = gr.Blocks()
|
|
|
|
with demo:
|
|
gr.Markdown(
|
|
r"Let's do some kinematics! Choose the speed and angle to see the trajectory. Remember that the range $R = v_0^2 \cdot \frac{\sin(2\theta)}{g}$"
|
|
)
|
|
|
|
with gr.Row():
|
|
speed = gr.Slider(1, 30, 25, label="Speed")
|
|
angle = gr.Slider(0, 90, 45, label="Angle")
|
|
output = gr.LinePlot(
|
|
x="x",
|
|
y="y",
|
|
overlay_point=True,
|
|
tooltip=["x", "y"],
|
|
x_lim=[0, 100],
|
|
y_lim=[0, 60],
|
|
width=350,
|
|
height=300,
|
|
)
|
|
btn = gr.Button(value="Run")
|
|
btn.click(plot, [speed, angle], output)
|
|
|
|
if __name__ == "__main__":
|
|
demo.launch()
|