python - 导入时未定义枚举对象 - 通过 Jupyter Notebook 的 Python 3

Question

我正在通过 Jupyter Notebook 使用 Python 3我在同一工作目录中有两个模块“color.py”和“painting.ipynb”。

颜色.py

from enum import Enum

class Color(Enum):
    RED = 1
    BLUE = 2

def painting(color):
        if color == Color.RED:
            print("The painting is red")
        elif color == Color.BLUE:
            print("The painting is blue")
        else:
            raise ValueError("The painting is not blue nor red!")

绘画.ipynb

import color 
color.painting(Color.RED)

当我尝试在 jupyter notebook 中运行“painting.ipynb”时，出现以下错误。

--------------------------------------------------------------------------- NameError Traceback (most recent call last) in () 1 import color 2 ----> 3 color.painting(Color.RED)

NameError: name 'Color' is not defined

我不明白为什么我无法访问颜色。我应该给类(class)打电话吗？这是我第一次使用 jupyter notebook 和 Enum。请帮忙 :) 谢谢。

Answer 1

因为您要导入整个 .py 文件；您需要使用以下内容来引用您的“颜色”枚举:

import color 
color.painting(color.Color.RED)

根据您计划引用该枚举的频率，单独导入您需要的位可能会更好:

from color import Color
from color import painting