无法引用函数外部声明的特定变量

时间:2014-07-30 15:04:56

标签: python python-2.7 matplotlib scope

我试图使用Python为射弹运动路径制作动画。为了达到这个目的,我使用了matplotlib的动画模块。我的完整脚本如下。

#!/usr/bin/env python

import math
import sys

import matplotlib.animation as anim
from matplotlib.pyplot import figure, show

# Gravitational acceleration in m/s/s
g = -9.81
# Starting velocity in m/s.
vel = 100
# Starting angle in radians.
ang = 45 * math.pi / 180
# Starting height in m.
y0 = 0
# Time settings.
t = 0
dt = 0.1
time = -vel**2 * math.sin(2 * ang) / g

def projectile():
    print g, vel, ang, y0, dt, time
    print t # Crashes here with UnboundLocalError: local variable 't' referenced before assignment
    t=0 # With this it works
    x = 0
    y = 0.1
    xc = []
    yc = []

    # Simulate the projectile.
    while t < time:
        x = vel * math.cos(ang) * t
        y = vel * math.sin(ang) * t + (g * t**2) / 2 + y0
        if y < 0:
            break
        t += dt
        xc.append(x)
        yc.append(y)
        yield x, y

def update_frame(sim):
    x, y = sim[0], sim[1]
    line.set_data(x, y)
    return line,

def init():
    line.set_data([], [])
    return line,

# Show the results.
fig = figure()
ax = fig.add_subplot(111)
ax.set_xlim([-5,1500])
ax.set_ylim([0,300])

# ax.plot returns a single-element tuple, hence the comma after line.
line, = ax.plot([], [], 'ro', ms=5)
ani = anim.FuncAnimation(fig=fig, func=update_frame, frames=projectile, blit=True, interval=20, init_func=init, repeat=False)
show()

我遇到的问题是我似乎能够使用除t之外的所有变量。我这样做是为了创建一个停止条件,所以它只会运行一次,后来我发现了repeat=False,但我仍然对此感到好奇。为什么我不能在t内使用projectile?我正在使用Anaconda 1.9.1软件包中的Matplotlib 1.3.1运行Python 2.7.6。

1 个答案:

答案 0 :(得分:3)

问题出现是因为您尝试重新分配全局变量t

您只能访问的变量g, vel, ang, y0, dt, time(无需重新分配),因此python尝试在本地访问它们,然后全局访问它们。但是,当您使用行t 重新分配 t += dt时,您实际上是在告诉python创建一个标识为t的局部变量,并为其指定所需的值。因此,无法访问全局标识t,因为您已告诉python t是本地的,当您在分配t之前尝试访问UnboundLocalError时,您t 1}}被提出。

要告诉python将global t重新分配为全局变量,您只需将t = 0 (..) def projectile(): print g, vel, ang, y0, dt, time global t # now t can be edited as a global variable print t #works x = 0 y = 0.1 xc = [] yc = [] while t < time: (..) t += dt 添加到函数的开头:

>>> l = [1,2,3]
>>> def changelist():
...    l.append(4)
...
>>> changelist()
>>> l
[1,2,3,4]

修改

正如 flebool 指出的那样,只要您不为其重新分配标识符,您实际上仍然可以更改全局变量。例如,以下代码有效:

{{1}}