如何在函数定义中强制自动参数转换?

2022-02-23 00:00:00 python types type-conversion

问题描述

考虑以下函数示例(及其用法),该函数接收一个只有int意义的参数:

def fun(index):
    index = int(index)
    # do something with index

def main():
    print fun(1e6)

我发现我在几个处理整数的函数上不断重复此模式,因为它确保我真正接收到某个整数(请注意,1e6是一个浮点实例)。

这有点麻烦,如果忘记了它可能会产生问题(有代码对浮点/整数/字符串/.的反应不同.值)。那么这能以某种方式编码到函数签名中吗?

例如,类似于此伪代码的内容:

def fun(int index):
    # do something with index

解决方案

在Python3中,您可以使用function annotations,然后编写一个将参数转换(或尝试转换)为您指定类型的修饰符。

在Python2中,您不能以您想要的方式完成此操作。Python2的语法不支持将除参数本身以外的任何内容放入函数定义中,并且您不能更改Python的语法以允许您将类型也放入函数定义中。

相关文章