return heights

参数设定

initial_height = 100 # 初始高度

bounce_ratio = 0.8 # 反弹高度比例

n_bounces = 10 # 反弹次数

调用函数

heights = simulate_bouncing_ball(initial_height, bounce_ratio, n_bounces)

输出结果

for i, h in enumerate(heights):

print(f第{i+1}次落地，反弹高度为: {h:.2f}米)

模型的扩展

通过上述代码，我们可以得到小球每次反弹的高度。如果我们将反弹比例进行调整，甚至引入一些随机因素，模型的复杂性也可以大大增加。考虑空气阻力或表面摩擦的影响，都会使得反弹高度有所不同。

数据可视化

为了更好地理解小球的运动轨迹，我们还可以通过可视化来展示反弹过程。像Matplotlib这样的库可以帮助我们生成小球运动的图形。

import matplotlib.pyplot as plt

可视化小球反弹过程

def plot_bouncing_ball(heights):

plt.figure(figsize=(10, 5))

plt.plot(range(len(heights)), heights, marker=o)

plt.title(小球反弹高度变化)

plt.xlabel(反弹次数)

plt.ylabel(高度 (米))