使用mypy检查类型提示时，Pandas Timedelta的unit参数传入字符串变量的类型兼容问题咨询

大家好，我最近写Python代码时碰到了一个mypy类型检查的小问题，想跟各位请教探讨下：

我给变量period标注了str类型，之后用它来初始化pandas.Timedelta对象的unit参数，代码运行起来完全正常，但mypy却抛出了类型不兼容的错误提示。

下面是我提炼的最小可复现示例：

import pandas as pd

period: str = "minute"
my_timedelta = pd.Timedelta(30, unit=period)

print(my_timedelta)

仔细排查后发现，问题出在pd.Timedelta的unit参数类型定义上——虽然我传入的字符串值是合法的时间单位，但mypy识别的unit参数类型其实是字面量联合类型（比如"nanoseconds" | "ns" | "microseconds" | "us" | ...这类具体的合法单位字符串），而不是宽泛的str类型，所以当我传入一个普通的str变量时，mypy就会判定类型不匹配。

给大家整理几个可行的解决办法：

• 方法一：用Literal类型标注变量
  从typing（Python 3.8+）或者typing_extensions导入Literal，把period的类型标注为具体的合法单位字面量，这样mypy就能识别它是符合要求的：

  from typing import Literal
  import pandas as pd
  
  period: Literal["minute", "min"] = "minute"
  my_timedelta = pd.Timedelta(30, unit=period)
  

• 方法二：用cast做类型转换
  如果不想限制period的字面量范围，可以用typing.cast把变量转换成pd.TimedeltaUnit类型（Pandas内置的单位类型），明确告诉mypy这个变量是合法的：

  from typing import cast
  import pandas as pd
  
  period: str = "minute"
  my_timedelta = pd.Timedelta(30, unit=cast(pd.TimedeltaUnit, period))
  

• 方法三：忽略该行的mypy检查
  这是最直接的临时方案，在代码行末尾加# type: ignore，但不建议长期使用，因为会掩盖潜在的类型问题：

  my_timedelta = pd.Timedelta(30, unit=period)  # type: ignore
  

如果各位有其他更优雅的解决方式，欢迎在评论区补充交流~

内容来源于stack exchange