迹忆客 专注技术分享

当前位置:主页 > 学无止境 > 编程语言 > Python >

如何在 Python 中使用枚举类型提示

作者:迹忆客 最近更新:2022/08/28 浏览次数:

使用枚举类在 Python 中键入提示枚举,例如 def get_value_from_enum(size: Sizes):。 然后,我们可以访问函数主体中枚举的任何成员而不会收到警告。

from enum import Enum


class Sizes(Enum):
    SMALL = 1
    MEDIUM = 2
    LARGE = 3


def get_value_from_enum(size: Sizes):
    print(size.name)  # 👉️ MEDIUM
    print(size.value)  # 👉️ 2

    return size.value


result = get_value_from_enum(Sizes.MEDIUM)

print(result)  # 👉️ 2

python 枚举类型限定

我们使用 Sizes 枚举类来键入提示枚举。

另一种方法是使用 Literal 类型来指定函数参数可以是枚举中的成员之一。

from typing import Literal
from enum import Enum


class Sizes(Enum):
    SMALL = 1
    MEDIUM = 2
    LARGE = 3


def get_value_from_enum(size: Literal[Sizes.SMALL, Sizes.MEDIUM, Sizes.LARGE]):
    print(size.name)  # 👉️ MEDIUM
    print(size.value)  # 👉️ 2

    return size.value


result = get_value_from_enum(Sizes.MEDIUM)

print(result)  # 👉️ 2

文字类型提示可用于向类型检查器指示函数参数的值等于提供的文字(或示例中的几个文字之一)。

转载请发邮件至 1244347461@qq.com 进行申请,经作者同意之后,转载请以链接形式注明出处

本文地址:

相关文章

Python 中的 Pandas 插入方法

发布时间:2024/04/23 浏览次数:112 分类:Python

本教程介绍了如何在 Pandas DataFrame 中使用 insert 方法在 DataFrame 中插入一列。

Pandas 重命名多个列

发布时间:2024/04/22 浏览次数:199 分类:Python

本教程演示了如何使用 Pandas 重命名数据框中的多个列。

扫一扫阅读全部技术教程

社交账号
  • https://www.github.com/onmpw
  • qq:1244347461

最新推荐

教程更新

热门标签

扫码一下
查看教程更方便