如何输入泛型函数？

Question

说我想在 Python 中定义一些东西与 Java 中的以下函数完全相同：

public <T> void func(T a, T b):
     ...

这段代码只是意味着 func 可以将任何 a 和 b 作为输入，只要它们具有相同的数据类型。

Python 本身没有显式类型检查，但现在我正在使用 mypy 检查我的代码。我的尝试如下：

T = TypeVar('T')

def func(a: Generic[T], b: Generic[T]):
         ...
func(1, 3)  # this should be a legal input since 1 and 3 are both int

和

T = TypeVar('T')

class A(Generic[T]):
    pass

def func(a: A[T], b: A[T]):
    ...
func(1, 3)

但是，mypy 报告了两者的错误。有没有办法让我们像Java一样定义这样的通用函数来通过mypy检查？我知道为了对我的代码进行完整性检查，我还可以在函数内部检查 type(a) 是否等于 type(b)，但这不是我的目的。我对我的问题特别感兴趣，即如何定义这样一个通用函数来通过 mypy 检查？

Answer 1

Java 函数实际上并没有按照您的意愿执行，Python 等价函数也不会按照您的意愿执行。

你想确保 Java 函数总是用两个相同类型的参数调用，但是 any 你可能传递给它的两个对象共享一个类型。具体来说，任何两个对象都是 Object 的实例。（null 和原语使事情变得有点复杂，但它们不会停止类型检查的调用。）任何调用都会进行类型检查，而您尝试使用泛型将无济于事。

同样，Python 等价于

def func(a: T, b: T):
    ...

和 Java 代码一样，任何调用都会进行类型检查，因为任何两个对象都是 object.

的实例

无法使用类型注释指定 type(a) is type(b) 的约束。无论如何，这通常是错误的约束。您通常想要指定的是参数在某种程度上相互兼容。不幸的是，也没有真正好的方法。

例如，如果您想在 Java 中指定两个参数相互比较，那就是

public <T extends Comparable<T>> void func(T a, T b) {...}

但是，Python 类型变量不支持这样的参数化边界。

如何输入泛型函数？

How to typehint a generic function?

python

python-3.x

mypy