写一个小球匀速运动的代码模型python

下面是一个简单的小球匀速运动的代码模型：

import pygame

pygame.init()

# 定义窗口大小
width, height = 600, 400
screen = pygame.display.set_mode((width, height))

# 定义小球的初始位置和速度
x, y = 50, 50
vx, vy = 5, 5

# 定义小球的半径和颜色
radius = 20
color = (255, 0, 0)

# 游戏循环
while True:
    # 处理退出事件
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            exit()

    # 小球移动
    x += vx
    y += vy

    # 判断是否碰到边界，如果碰到则修改速度方向
    if x - radius < 0 or x + radius > width:
        vx = -vx
    if y - radius < 0 or y + radius > height:
        vy = -vy

    # 填充背景色
    screen.fill((255, 255, 255))

    # 画小球
    pygame.draw.circle(screen, color, (x, y), radius)

    # 更新屏幕显示
    pygame.display.update()

在这个模型中，我们使用了 Pygame 库来创建窗口和绘制图形。我们定义了小球的初始位置和速度，并且在游戏循环中不断地更新小球的位置。如果小球碰到了窗口的边界，我们就修改小球的速度方向。最后，我们使用 Pygame 的绘图函数来画出小球，并且更新屏幕的显示。

你可以根据自己的需求修改代码中的参数，例如窗口大小、小球的半径和速度等等。