我需要输入水平初速度和小球高度

好的，你可以将代码中的初始速度 `v0` 和初始高度 `h0` 替换成你需要的值。例如，如果你想要将初始速度设为 20 m/s，初始高度设为 2 m，那么代码如下：

import matplotlib.pyplot as plt
import numpy as np

# 设置初始速度和高度
v0 = 20.0          # 初始速度，单位 m/s
h0 = 2.0           # 初始高度，单位 m

# 设置重力加速度
g = 9.8            # 重力加速度，单位 m/s^2

# 计算小球运动时间
t_max = 2.0 * v0 / g
t = np.linspace(0, t_max, 51)

# 计算小球的水平和竖直位移
x = v0 * t
y = h0 + v0 * t - 0.5 * g * t**2

# 绘制小球轨迹
plt.figure(figsize=(6, 4.5))
plt.axis([0, 25, 0, 5])
plt.xticks(fontsize=14)
plt.yticks(fontsize=14)
plt.xlabel('水平位移 (m)', fontsize=14)
plt.ylabel('竖直位移 (m)', fontsize=14)
for i in range(len(t)):
    plt.plot(x[i], y[i], 'ro')
    plt.plot(x[:i+1], y[:i+1], 'r--')
    plt.pause(0.1)
plt.show()

这样，你就可以输入你需要的水平初速度和小球高度，得到对应的轨迹图像了。