Python3.5 版本引入了类型提示(Type Hints),它允许开发者在代码中显式地声明变量、函数、方法等的类型信息。这种类型声明不会影响 Python 解释器的运行,但可以让 IDE 和静态分析工具更好地理解代码,同时提高代码的可读性和可维护性。然而,由于 Python 支持动态类型,类型提示并不能完全确保代码的正确性。本文仅介绍 Python 类型提示的初步使用。如果需要更详细的使用说明,请参考以下文章:typing、Python 类型提示简介和Type Hints 入门教程。
类型提示的语法格式为:
- 对于变量:{变量名}:{类型名} =
- 对于函数参数:{参数名}:{类型名} =
- 对于函数返回值:->
- 1 类型提升
-
2 类型声明
- 2.1 基本类型
- 2.2 嵌套类型
- 2.3 自定义类型
-
2.4 复合类型
- 2.4.1 Union和Optional
- 2.4.2 Generator和Iterator
- 2.4.3 Callable
- 2.4.4 Any和NoReturn
- 2.4.5 其他
- 2.5 类型提示的别名
- 3 参考
1 类型提升
类型提示的引入主要有以下几个方面的用途:
1 提高代码可读性
类型提示可以帮助其他开发人员更好地理解代码,特别是在处理大型代码库时。通过清晰地指定变量、函数参数和返回值的数据类型,开发人员可以更快地理解代码的含义和用途,从而更容易维护和修改代码。
如下所示。我们有一个名为 add 的函数,用于将两个数字相加并返回结果。以下是该函数的原始代码:
def add(a, b):
return a + b
我们发现,该函数没有任何类型提示,因此在调用该函数时,我们必须自己去了解和检查每个参数的类型。这样会导致代码的可读性和可维护性变差,特别是在代码规模较大、涉及多个文件的情况下。为了改善这种情况,我们可以使用类型提示来明确指定每个参数的类型。以下是添加类型提示后的 add
函数的代码:
def add(a: int, b: int) -> int:
return a + b
现在,我们可以清楚地看到函数 add 的参数和返回值都是整数类型。这使得代码更易于理解,也提高了代码的可靠性。
2 检测类型错误
Python 是一种动态语言,因此变量和函数参数的类型可以在运行时进行更改。但是,这也意味着开发人员容易在代码中引入类型错误。通过使用类型提示,开发人员可以在编译时检测到这些类型错误,并更早地发现和修复它们,从而减少代码错误和调试时间。
mypy是一个用于检查Python类型的静态类型检查器。它可以检测类型注释中的错误以及其他类型的错误。mypy使用说明可以参考:mypy简易教程。mypy需要首先输入以下命令安装:
pip install mypy
然后,在代码中标注变量、函数参数和返回值的类型。运行以下命令:
mypy your_script.py
在上面的示例中,your_script.py是要检查的Python脚本。运行mypy工具后,它将检查Python脚本中的类型错误,并输出错误信息。
3 提供自动补全和文档
许多集成开发环境(IDE)和编辑器都可以使用类型提示来提供自动补全和代码文档。这可以帮助开发人员更快地编写代码,并提供关于函数参数和返回值的信息,以便更好地理解代码。要使用Python类型提示提供自动补全和文档,需要使用一个支持该功能的Python编辑器。比如一些流行的Python编辑器包括vscode、PyCharm和Sublime Text等。
以vscode为例,考虑一个整数相加函数,将结果保存在变量c中。如果加上类型提示,vscode插件将推断变量c的类型为 int,并提供代码补全和代码提示等功能。
此外,还可以使用vscode的autoDocstring生成带有类型提示的文档和注释。
autoDocstring注释代码使用方法如下所示:
按照以上方法,对于有无类型提示的注释结果如下:
def add(a, b):
"""_summary_
Args:
a (_type_): _description_
b (_type_): _description_
Returns:
_type_: _description_
"""
c = a + b
return c
def add(a: int, b: int) -> int:
"""_summary_
Args:
a (int): _description_
b (int): _description_
Returns:
int: _description_
"""
c = a + b
return c
2 类型声明
2.1 基本类型
对于Python的内置基本类型 int、float、str 和 byte等,可以直接使用类型本身进行类型提示。如下所示:
# 直接定义
age: int = 1
# 声明后定义
num: float
num = 2.0
def greet(name: str) -> str:
return f"Hello, {name}!"
def is_even(x: int) -> bool:
return x % 2 == 0
def encode_data(data: str) -> bytes:
return data.encode('utf-8')
2.2 嵌套类型
对于容器数据结构,例如 list、tuple、dict 等,也可以直接使用类型本身进行类型提示。如下所示:
items: list = [1, 4.0, "3"]
info: dict = {"name":"john", "age":24}
在Python的容器数据结构中,每个元素都具有其自己的类型。虽然这种方法提供了灵活性,但是内部元素的类型无法受到限制,因此内部元素可以是任何类型(Any)。可以通过Python的typing标准库来声明这些类型及其元素类型。
from typing import List, Tuple, Dict, Set
# 指定my_list变量是一个整数列表
my_list: List[int] = [1, 2, 3, 4]
# 指定my_tuple变量应该是一个按顺序包含整数、字符串和布尔值的元组
my_tuple: Tuple[int, str, bool] = (1, "hello", True)
# 指定了my_dict变量是一个所有键为str类型,所有值为int类型的字典
my_dict: Dict[str, int] = {"apple": 1, "banana": 2, "orange": 3}
# 指定了my_set变量应该是一个浮点数集合
my_set: Set[float] = {1.0, 2.0, 3.0}
2.3 自定义类型
Python也支持对自定义类进行类型提示。下面是一个自定义类的类型提示示例:
class Person:
def __init__(self, name: str, age: int):
self.name = name
self.age = age
def say_hello(person: Person) -> str:
return f"Hello, {person.name}!"
在上面的代码中,我们定义了一个 Person 类,它有两个属性:name 和 age。在初始函数中,我们使用类型提示指定了这两个属性的类型。接下来,我们定义了一个 say_hello 函数,这个函数的参数是一个 Person 类型的对象,并且返回值是一个字符串。
对于numpy和pandas这种第三方库,也可以通过同样的方法进行类型提示:
import numpy as np
import pandas as pd
import cv2
# numpy
def add_arrays(a: np.ndarray, b: np.ndarray) -> np.ndarray:
return np.add(a, b)
# pandas
def filter_dataframe(df: pd.DataFrame, column: str, value: float) -> pd.DataFrame:
return df[df[column] > value]
# opencv,opencv图像本身就是一个numpy数组结构
def resize_image(img: np.ndarray, height: int, width: int) -> np.ndarray:
return cv2.resize(img, (width, height))
2.4 复合类型
2.4.1 Union和Optional
Python的typing库也提供了Union类型用于表示多种类型中的一种,Optional类型用于表示可选类型。它们可以结合使用,以便更好地表示变量的类型。
例如,如果一个变量可以是整数或字符串类型,那么可以这样定义它的类型:
from typing import Union
def func(x: Union[int, str]) -> None:
pass
上面的代码中,x的类型为Union[int, str],表示x可以是整数或字符串类型。
如果一个变量可以是整数类型或None类型,那么可以这样定义它的类型:
from typing import Optional
def func(x: Optional[int] = None) -> None:
pass
Union和Optional类型可以结合使用。例如,如果一个变量可以是整数类型、字符串类型或None类型,那么可以这样定义它的类型:
from typing import Optional, Union
def func(x: Optional[Union[int, str]]) -> None:
pass
上面的代码中,x的类型为Optional[Union[int, str]],表示x可以是整数类型、字符串类型或None类型。
此外,在Python中,Union[X, Y] 表示变量的类型可以是 X 或 Y。因此,Optional[X] 实际上是 Union[X, None] 的简写形式。这种语法的好处是它可以使代码更简洁,因为我们只需要写一个类型而不是两个。
from typing import Optional, Union
def greet(name: Optional[str]) -> str:
if name is None:
return "Hello, stranger!"
else:
return f"Hello, {name}!"
def greet2(name: Union[str, None]) -> str:
if name is None:
return "Hello, stranger!"
else:
return f"Hello, {name}!"
在上面代码中,greet和greets函数是等价的。在第一个函数中,我们使用了 Optional[str] 来表示 name 可以是一个字符串或者是 None。在第二个函数中,我们使用了 Union[str, None] 来达到相同的效果。
2.4.2 Generator和Iterator
在Python中,Generator和Iterator是非常常见的数据类型。Generator是一种函数,可以通过yield语句生成一个迭代器,而Iterator是一种对象,可以用于迭代元素序列。为了提高代码的可读性和可维护性,我们可以使用类型提示来指定Generator和Iterator的类型。
Generator类型提示使用Generator[ReturnType, SendType, ReturnType]语法,其中ReturnType指定返回值类型,SendType指定发送值类型,ReturnType指定生成器的类型。例如,下面是一个简单的Generator类型提示示例:
from typing import Generator
def even_numbers(n: int) -> Generator[int, None, None]:
for i in range(n):
if i % 2 == 0:
yield i
上面的代码中,even_numbers是一个Generator函数,返回类型是Generator[int, None, None],该函数生成一个整数序列,其中每个偶数都是通过yield语句生成的。
Iterator类型提示使用Iterator[ElementType]语法,其中ElementType指定迭代器元素类型。例如,下面是一个简单的Iterator类型提示示例:
from typing import Iterator
class MyIterator:
def __init__(self):
self.current: int = 0
self.max: int = 5
def __iter__(self) -> Iterator[int]:
return self
def __next__(self) -> int:
if self.current >= self.max:
raise StopIteration
else:
self.current += 1
return self.current
在上面的代码中,我们对MyIterator类进行了注释。使用了typing模块中的Iterator类来注释__iter__()
方法的返回值类型。同时,我们对current和max属性也进行了注释,指定了它们的类型为int。在__next__()
方法中,我们指定了返回值类型为int。
2.4.3 Callable
Callable类型提示用于表示一个可调用对象,例如函数、类或对象等。从形式上来看,Callable类型提示接受两个或三个类型提示参数:第一个参数表示函数的参数类型,第二个参数表示函数的返回类型。下面是一个Callable类型提示的例子:
from typing import Callable
def apply(func: Callable[[int, int], int], a: int, b: int) -> int:
return func(a, b)
def add(a: int, b: int) -> int:
return a + b
result = apply(add, 3, 4)
print(result) # 输出7
在上面的例子中,apply函数接受一个名为func的参数,该参数是一个Callable类型,它指定了函数的两个整数参数和一个整数返回值。add函数满足这个条件,因此可以传递给apply函数,它会返回add(3, 4)的结果7。
2.4.4 Any和NoReturn
Any类型表示一个任意类型,它可以用于函数参数、函数返回值和变量等。使用Any类型时,我们可以省略类型注释,使变量类型更加灵活。下面是一个使用Any类型的例子:
from typing import Any
def print_value(value: Any) -> None:
print(value)
print_value("Hello World") # 输出 "Hello World"
print_value(123) # 输出 123
在上面的例子中,我们定义了一个print_value函数,它接受一个任意类型的参数value,并将其打印出来。我们可以看到,我们可以将任何类型的值传递给print_value函数,包括字符串和整数。这使得我们的代码更加灵活。
NoReturn类型表示函数不会返回任何值。这个类型通常用于标识那些没有返回值的函数。下面是一个使用NoReturn类型的例子:
from typing import NoReturn
def print_message(message: str) -> NoReturn:
print(message)
raise Exception("Error occurred")
print_message("Hello World") # 输出 "Hello World",然后抛出异常
在上面的例子中,我们定义了一个print_message函数,它接受一个字符串类型的参数message,并将其打印出来。然后,我们手动抛出了一个异常,这意味着函数不会返回任何值。我们可以使用NoReturn类型来明确地表示这一点。
2.4.5 其他
Python还支持更高级的类型提示。例如,可以使用Sequence来指定一个列表,使用TypedDict来指定一个带有特定键和值类型的字典。此外,Python还支持Literal类型提示,可以限制变量只能取特定的常量值。最近,Python3.8版本还增加了Protocol类型提示,允许指定类需要实现哪些方法和属性。这些类型提示用的不多,但是如果需要更精细的类型控制,可以参考官方文档:typing。
2.5 类型提示的别名
在类型提示中使用了过于复杂的类型,可以考虑将其定义为一个类型别名,然后在函数参数、返回值等处使用该类型别名。例如,如果你需要传递一个包含多个字段的字典作为函数参数,你可以使用Dict[str, Union[int, str, List[int]]]来表示该字典的类型。但是,这个类型过于复杂,不易于理解。你可以将其定义为一个类型别名,如下所示:
from typing import Dict, Union, List
MyDict = Dict[str, Union[int, str, List[int]]]
def my_function(my_dict: MyDict) -> int:
# Function body
return 1
这样,你就可以在函数参数、返回值等处使用MyDict
这个类型别名,使代码更加易读、易懂。
3 参考
- typing
- Python 类型提示简介
- Type Hints 入门教程
- mypy
- mypy简易教程
- autoDocstring
服务器托管,北京服务器托管,服务器租用 http://www.fwqtg.net
机房租用,北京机房租用,IDC机房托管, http://www.e1idc.net