如何为仅接受np.sin或np.cos的函数参数添加正确类型提示？

你遇到的问题是因为Literal类型仅支持字面量（如字符串、数字、布尔值等），而np.sin和np.cos是numpy的ufunc实例（函数对象），不属于字面量范畴，所以类型检查工具会报错。下面是几种可行的正确写法：

方案1：用TypeVar限定特定函数对象 #

通过TypeVar绑定np.sin和np.cos，让类型检查工具严格限制参数只能是这两个具体函数：

import numpy as np
from typing import TypeVar

# 定义仅允许np.sin和np.cos的类型变量
SinOrCos = TypeVar('SinOrCos', np.sin, np.cos)

def foo(f: SinOrCos):
    # 示例实现
    return f(np.array([0, np.pi/2]))

方案2：兼容新版类型检查工具的Literal写法 #

如果你的类型检查工具（比如mypy 0.910+、pyright）支持非字面量的Literal使用，可以直接保留Literal写法，但需要确保工具版本足够新：

import numpy as np
from typing import Literal

def foo(f: Literal[np.sin, np.cos]):
    ...

如果仍报错，可尝试给mypy添加--allow-any-expr-in-literal启动选项。

方案3：通用类型+注释（兼容旧工具） #

如果使用的是旧版类型检查工具，无法支持前两种方案，可以先指定参数为np.ufunc类型，再通过文档注释明确允许的具体函数：

import numpy as np

def foo(f: np.ufunc):
    """
    Args:
        f: 仅接受np.sin或np.cos函数
    """
    ...

这种方式兼容性最好，但类型检查工具无法强制限制，仅起到文档提示作用。

内容的提问来源于stack exchange，提问作者schtandard