我这个程序的最终目标是用弹簧模拟受约束的陀螺,但这是以后的目标,目前我一直在努力了解对象创建、定位和交互如何工作,并最终建立起来.
目前,我一直在try 让我的代码并排产生N个独立的课程,但它们都在同一位置产生.
我try 过很多事情,从try 定义一次创建多个课程的方法,到当前的迭代,我try 使用for循环来创建和移动钟摆的位置,但没有效果.
import numpy as np
import manim as mn
class Pendulum:
def __init__(self, mass, length, theta):
self.mass = mass
self.length = length
self.g = -9.81
self.angle = theta
self.angular_vel = 0
def step(self, dt):
# Defining RK4
def runge_kutta_4(f, t, y0, h=None):
if h is None:
h = t[1] - t[0]
n = len(t)
y = np.zeros((n, len(y0)))
y[0] = y0
for i in range(n - 1):
k1 = h * f(t[i], y[i])
k2 = h * f(t[i] + h/2, y[i] + k1/2)
k3 = h * f(t[i] + h/2, y[i] + k2/2)
k4 = h * f(t[i] + h, y[i] + k3)
y[i+1] = y[i] + (k1 + 2*k2 + 2*k3 + k4) / 6
return y
def pendulum_equations(t, state):
theta, omega = state
force = self.mass * self.g * np.sin(theta)
torque = force * self.length
MoI = 1/3 * self.mass * self.length**2
alpha = torque / MoI
return np.array([omega, alpha])
state = np.array([self.angle, self.angular_vel])
t = np.array([0, dt])
sol = runge_kutta_4(pendulum_equations, t, state)
self.angle = sol[-1, 0]
self.angular_vel = sol[-1, 1]
class PhysicalPendulum(mn.Scene):
def construct(self):
p = Pendulum(2, 10, np.pi/2)
N = 3 # change number of pendulums here
pendulums = []
scale = 0.5
spacing = 3 # Adjust the spacing between pendulums as needed
def get_pendulum(i, rod_width=0.2, rod_height=1):
rod = mn.Rectangle(width=rod_width, height=scale * p.length, color=mn.BLUE)
rod.shift(mn.DOWN * scale * p.length / 2)
rod.rotate(p.angle, about_point=rod.get_top())
pendulum = mn.VGroup(rod)
pendulum.shift(mn.UP * 3) # Adjust the vertical shift as needed
if i % 2 == 0:
pendulum.shift(mn.RIGHT * spacing * i)
else:
pendulum.shift(mn.LEFT * spacing * i)
return pendulum
def step(pendulum, dt, i):
p.step(dt)
pendulum.become(get_pendulum(i))
for i in range(N):
pendulum = get_pendulum(i)
pendulum.add_updater(lambda mob, dt: step(mob, dt, i))
pendulums.append(pendulum)
self.add(*pendulums)
self.wait(20)
for pendulum in pendulums:
pendulum.remove_updater(step)
这也是我第一次try 面向对象编程,所以我很感激有关如何改进编码风格的任何提示以及对我做错的事情的任何 comments .