根据函数中的参数数量返回不同的值

时间:2018-01-27 10:24:31

标签: python python-3.x

我试图创建一个从1到5个参数的函数,并根据给定的数字进行不同的计算。 我的想法是这样的:

def function(*args)
    num_of_args = (!!here is the problem!!)
if(num_of_args == 1) : result = a
else if(number_of_args == 2) : result = a+b

等等 我试图计算参数的数量并将该数字分配给变量,但无法找到方法 我想可能没有必要使用5,但在我设法计算这些参数之前,我真的不想专注于它

2 个答案:

答案 0 :(得分:3)

您可以使用len(args)

def function(*args):
    if len(args) == 0:
        print("Number of args = 0")
    elif len(args) == 1:
        print("Number of args = 1")
    else:
        print("Number of args >= 2")

答案 1 :(得分:2)

*args,位置论证以列表形式发送。您可以使用args [0],args [1] ....以及len(args)的长度来访问它们

def foo(*args):
  print(len(args))