下面是一个使用Python和Pygame库的示例代码，实现了按键时移动矩形的功能：

import pygame
from pygame.locals import *

# 初始化Pygame库
pygame.init()

# 设置窗口大小和标题
width, height = 640, 480
screen = pygame.display.set_mode((width, height))
pygame.display.set_caption("移动矩形")

# 定义矩形的初始位置和大小
rect_x, rect_y = 50, 50
rect_width, rect_height = 50, 50

# 游戏主循环
running = True
while running:
    # 清空屏幕
    screen.fill((0, 0, 0))

    # 绘制矩形
    pygame.draw.rect(screen, (255, 255, 255), (rect_x, rect_y, rect_width, rect_height))

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

    # 处理事件
    for event in pygame.event.get():
        if event.type == QUIT:
            running = False
        elif event.type == KEYDOWN:
            # 根据按键移动矩形
            if event.key == K_LEFT:
                rect_x -= 10
            elif event.key == K_RIGHT:
                rect_x += 10
            elif event.key == K_UP:
                rect_y -= 10
            elif event.key == K_DOWN:
                rect_y += 10

# 退出游戏
pygame.quit()

在这个示例中，我们使用Pygame库创建了一个窗口，并在窗口中绘制一个矩形。通过监听键盘事件，我们可以根据按键来移动矩形的位置。按下向左、向右、向上和向下的箭头键时，矩形会分别向左、向右、向上和向下移动10个像素。

你可以将以上代码保存为一个.py文件，在安装了Pygame库的Python环境中运行，即可看到按键时移动矩形的效果。