有没有一种简单而有效的方法来编写python中的跳转和重力逻辑，使用pygame作为平台？

import pygame
from pygame.locals import *

pygame.init()

screen = pygame.display.set_mode((width, height))

gravity = 0.5
jump_force = 10

class Player(pygame.sprite.Sprite):
    def __init__(self):
        pygame.sprite.Sprite.__init__(self)
        self.image = pygame.Surface((width, height))
        self.rect = self.image.get_rect()
        self.rect.x = x
        self.rect.y = y
        self.velocity = 0

def update(self):
    self.velocity += gravity
    self.rect.y += self.velocity
    if self.rect.y >= ground_y:
        self.rect.y = ground_y
        self.velocity = 0

def jump(self):
    self.velocity -= jump_force

running = True
while running:
    for event in pygame.event.get():
        if event.type == QUIT:
            running = False
        elif event.type == KEYDOWN:
            if event.key == K_SPACE:
                player.jump()
    
    player.update()
    screen.fill((0, 0, 0))
    screen.blit(player.image, player.rect)
    pygame.display.flip()