在PyGame中创建Ball类对象列表时遇UnboundLocalError求助

错误根源 #

你遇到的UnboundLocalError是因为变量名冲突：在run()函数的绘制循环里，你用了for Ball in balls:，这里的Ball是局部变量，覆盖了全局作用域的Ball类。Python解释器会认为你在balls.append(Ball(...))里引用的是这个还未赋值的局部变量，因此抛出错误。

完整修复代码 #

import pygame
import math
 
pygame.init()
 
# 屏幕设置
framespd = 30
gravity = 1
t = 0.01
clock = pygame.time.Clock()
width, height = 1000, 1000
pygame.display.set_caption("Orbit Take 1")
screen = pygame.display.set_mode([width, height])
 
class Ball:
    def __init__(self, angle, velocity, radius, color, x_pos, y_pos):
        # 转换输入为正确的数据类型
        self.angle = math.radians(float(angle))  # 角度转弧度
        self.velocity = float(velocity)
        self.radius = int(radius)
        self.color = color
        self.x_pos = float(x_pos)
        self.y_pos = float(y_pos)
        # 初始化速度分量
        self.vel_x = self.velocity * math.cos(self.angle)
        self.vel_y = self.velocity * math.sin(self.angle)
 
    def draw(self):
        # 使用全局已创建的screen，无需重复传参
        pygame.draw.circle(screen, self.color, (int(self.x_pos), int(self.y_pos)), self.radius)
 
    def update_position(self):
        # 更新竖直方向速度（受重力影响）
        self.vel_y += gravity * t
        # 更新位置
        self.x_pos += self.vel_x * t
        self.y_pos += self.vel_y * t
 
balls = [] 
 
def run():
    global balls  # 声明使用全局的balls列表
    value = True
    # 只在程序启动时询问一次球的数量
    number_balls = int(input("How many balls: "))
 
    while len(balls) < number_balls:
        # 获取用户输入并转换类型
        a = input("Angle (degrees): ")
        v = input("Velocity: ")
        r = input("Radius: ")
        c = input("Color (e.g., 'red' or (255,0,0)): ")
        # 处理颜色输入：如果是RGB元组字符串，转换为实际元组
        if c.startswith('(') and c.endswith(')'):
            c = tuple(map(int, c.strip('()').split(',')))
        x_pos = '0'
        y_pos = '500'
        # 创建Ball对象并加入列表
        balls.append(Ball(a, v, r, c, x_pos, y_pos))
 
    while value:
        clock.tick(framespd)
        screen.fill('black')    
 
        # 更新所有球的位置并绘制
        for ball in balls:  # 用小写ball作为循环变量，避免覆盖类名
            ball.update_position()
            ball.draw()
 
        # 事件处理
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                value = False
        # 统一刷新屏幕
        pygame.display.flip()
    pygame.quit()       
 
run()

关键修复点 #

• 解决变量名冲突：将绘制循环的for Ball in balls:改为for ball in balls:，避免局部变量覆盖全局的Ball类。
• 输入类型转换：用户输入的字符串必须转换为数值类型（角度转弧度、速度/位置转浮点数、半径转整数），否则后续计算会报错。
• 优化绘制逻辑：Ball.draw()方法直接使用全局已创建的screen对象，无需在调用时重复创建窗口；将pygame.display.flip()移到事件循环外，确保所有绘制完成后统一刷新屏幕。
• 调整输入时机：将球数量的输入移到主循环外，避免每帧都弹出输入框干扰程序运行。
• 完善物理逻辑：实现了update_position()方法，处理重力对速度的影响并更新球的位置，让球能正常运动。

内容的提问来源于stack exchange，提问作者Amelia