pygame检测哪个键被按下

import pygame
from pygame.locals import *

pygame.init()

# 创建一个窗口
screen = pygame.display.set_mode((640, 480))

running = True

while running:
    for event in pygame.event.get():
        if event.type == QUIT:
            running = False
        elif event.type == KEYDOWN:
            if event.key == K_UP:
                print("Up key pressed")
            elif event.key == K_DOWN:
                print("Down key pressed")
            elif event.key == K_LEFT:
                print("Left key pressed")
            elif event.key == K_RIGHT:
                print("Right key pressed")

pygame.quit()