如何在 Python 中键入提示枚举
使用枚举类在 Python 中键入枚举提示。
然后,我们可以在函数体中访问枚举的任何成员,而不会收到警告。
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
我们使用 Sizes 枚举类来键入枚举提示。
使用 Literal 类型代替
另一种方法是使用 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
Literal
提示可用于向类型检查器指示函数参数的值等于提供的文字(或示例中的几个文字之一)。
相关文章
Pandas DataFrame DataFrame.shift() 函数
发布时间:2024/04/24 浏览次数:133 分类:Python
-
DataFrame.shift() 函数是将 DataFrame 的索引按指定的周期数进行移位。
Python pandas.pivot_table() 函数
发布时间:2024/04/24 浏览次数:82 分类:Python
-
Python Pandas pivot_table()函数通过对数据进行汇总,避免了数据的重复。
Pandas read_csv()函数
发布时间:2024/04/24 浏览次数:254 分类:Python
-
Pandas read_csv()函数将指定的逗号分隔值(csv)文件读取到 DataFrame 中。
Pandas 多列合并
发布时间:2024/04/24 浏览次数:628 分类:Python
-
本教程介绍了如何在 Pandas 中使用 DataFrame.merge()方法合并两个 DataFrames。
Pandas loc vs iloc
发布时间:2024/04/24 浏览次数:837 分类:Python
-
本教程介绍了如何使用 Python 中的 loc 和 iloc 从 Pandas DataFrame 中过滤数据。
在 Python 中将 Pandas 系列的日期时间转换为字符串
发布时间:2024/04/24 浏览次数:894 分类:Python
-
了解如何在 Python 中将 Pandas 系列日期时间转换为字符串