如何修复我分配的未分配变量的 UnboundLocalError？

Question

我正在尝试使用 Turtle 库和递归在 Python 中创建 Koch 雪花生成器。我有一个包含十六进制代码的字符串列表，我使用变量 color 让笔按照列表的顺序更改颜色。正如您所看到的，我在函数定义之外定义了 color ，但是当我尝试运行它时，我收到 UnboundLocalError ，指出 color 在分配之前已被引用。如何修复错误以便我的代码能够按预期运行？

这是我编写的一些用于生成雪花的代码。每次绘制一条线时，color 都会增加一，因此下一次绘制线时，使用的颜色是 palette[color % 6]

import turtle

turtle.colormode(255)
palette = ["#ff0f7b","#fd445d","#fc5552","#fa8139","#f98a34","#f89b29"]
color = 0

# Draw a Koch curve
def koch(iteration, length):
    if iteration == 1:
        turtle.forward(length/3)
        turtle.pencolor(palette[color % 6])
        color += 1
    else:
        koch(iteration-1, length/3)
    turtle.left(60)

这里是我收到的错误消息：

Traceback (most recent call last):
  File "C:\Users\hummi\Python Files\koch_snowflake.py", line 49, in <module>
    koch_snowflake(4, 480)
  File "C:\Users\hummi\Python Files\koch_snowflake.py", line 41, in koch_snowflake
    koch(iteration, length)
  File "C:\Users\hummi\Python Files\koch_snowflake.py", line 15, in koch
    koch(iteration-1, length/3)
  File "C:\Users\hummi\Python Files\koch_snowflake.py", line 15, in koch
    koch(iteration-1, length/3)
  File "C:\Users\hummi\Python Files\koch_snowflake.py", line 15, in koch
    koch(iteration-1, length/3)
  File "C:\Users\hummi\Python Files\koch_snowflake.py", line 12, in koch
    turtle.pencolor(palette[color % 6])
UnboundLocalError: local variable 'color' referenced before assignment

Answer 1

您只需要添加一个全局颜色语句，如下所示：

color = 0

# Draw a Koch curve
def koch(iteration, length):
    global color
    if iteration == 1:
        turtle.forward(length/3)
        turtle.pencolor(palette[color % 6])
        color += 1

如果没有添加，修改color的值将使其成为本地范围的变量。由于在该局部范围内，它尚未被分配，并且增量语句需要变量的先前值，因此您会收到所看到的错误。添加global color 语句会强制该语句引用全局color 值，即使它正在修改该值。如果您仅读取color，则不需要global color 语句。这里有一个有点奇怪的 Python 事实。

Answer 2

我的倾向是循环颜色，而不是使用全局索引和模块化算术：

import turtle
from itertools import cycle

# palette = cycle(["#ff0f7b", "#fd445d", "#fc5552", "#fa8139", "#f98a34", "#f89b29"])
palette = cycle(['#ff0000', '#000000', '#00ffff', '#0000ff', '#00ff00'])

# Draw a Koch curve
def koch(iteration, length):

    if iteration == 1:
        turtle.pencolor(next(palette))
        turtle.forward(length)
    else:
        koch(iteration-1, length/3)
        turtle.left(60)
        koch(iteration-1, length/3)
        turtle.right(120)
        koch(iteration-1, length/3)
        turtle.left(60)
        koch(iteration-1, length/3)

koch(4, 300)

turtle.done()

更改调色板以使颜色变化更加明显：