用*args将毕达哥拉斯定理推广到n个事物

Question

我了解到，当您不确定有多少参数可以传递给我的函数时，通常使用*arg参数。我正在尝试创建一个推广的毕达哥拉斯定理，这样它就可以计算n个事物，其中

pytha(*arg)

使用

print(pytha(x,y)) = np.sqrt(x**2 + y**2)
print(pytha(x,y,z)) = np.sqrt(x**2 + y**2 + z**2)
print(pytha(x,z)) = np.sqrt(x**2 + z**2)
print(pytha(x-7,y)) = np.sqrt((x-7)**2 + y**2)
print(pytha(x-3,y-5,z-8)) = np.sqrt((x-3)**2 + (y-5)**2 + (z-8)**2)
print(pytha(x,y,z,t)) = np.sqrt(x**2 + y**2 + z**2 + t**2)

我已经做了

def pytha(*arg):

但是我不知道如何在函数体中操作*arg。那么如何使用*arg创建这个函数呢？

Answer 1

您提供的参数将以您在函数定义期间提供的名称存储在元组中。如果它是*arg，那么元组将是arg，然后您可以将其与该名称一起使用。

>>> import numpy as np
>>> def pytha(*arg):
...     v = [i**2 for i in arg]   # collected all in one list
...     return np.sqrt(sum(v))    # give argument by summing the list items
... 
>>> pytha(2,3,4)
5.385164807134504

Answer 2

下面的解决方案与@Vicrobot的解决方案基本相同，不使用numpy。您可以遍历arg并计算每个元素的平方值，然后取这些值之和的平方根：

import math

def pytha(*arg):
    return math.sqrt(sum(elem**2 for elem in arg))