如何为函数中缺少的变量赋值零

Question

首先，感谢您的阅读。我是Python新手，每天都在学习新的东西。 我写了一个函数，其中输入是4个变量，输出是4个变量。 我的问题是变量是在其他地方定义的，它正在调用脚本。现在，如果用户只定义了3个变量呢？在这种情况下，我自动希望第4个变量被赋值为零值。我还包括了另一个代码，我正试图用于相同的目的

volsum = vol1 + vol2 + vol3 + vol4

    if vol1n == missing:
       vol1n = 0
    else:
    vol1n=vol1/volsum

    if vol2n == missing:
       vol2n = 0
    else:
    vol2n=vol2/volsum

    if vol3n = missing:
       vol3n = 0
    else:
    vol3n=vol3/volsum

    if vol4n == missing:
       vol4n = 0
    else:
    vol4n=vol4/volsum       

或者使用函数

def vol(vol1,vol2,vol3,vol4):
    volsum = vol1 + vol2
    vol1n=vol1/volsum
    vol2n=vol2/volsum
    vol3n=vol3/volsum
    vol4n=vol4/volsum
    return vol1n, vol2n,vol3n,vol4n

Answer 1

与其说是0，还可以忘记吗？就我个人而言，我不喜欢在不需要的时候打印出来。试试*args功能

def vol_func(*args):
    volumesum = sum(args)
    vals = []
    for i in range(len(args)):
        vals.append(args[i] / volumesum)
    return vals  

volsumlst = vol_func(2.3, 4)

Answer 2

定义功能时，请尝试指定值：

def vol(vol1 = 0,vol2 = 0,vol3 = 0,vol4 = 0):
    volsum = vol1 + vol2
    vol1n = vol1 / volsum
    vol2n = vol2 / volsum
    vol3n = vol3 / volsum
    vol4n = vol4 / volsum
    return vol1n, vol2n,vol3n,vol4n

现在假设您想跳过vol2：

variable_assigned = vol(vol1 = 5,vol3 = 2,vol4 = 5)

如果您需要一行（我喜欢创建一行）：

#defining vol()
def vol(*kwargs): return [i/sum(kwargs) for i in kwargs]
# using vol()
#example
vol (1,2,3)
#returns
[0.16666666666666666, 0.3333333333333333, 0.5]
# so to assign variables,
#if you give 3 values,
vol1n, vol2n, vol3n = vol(1,2,3)
# if 4,
vol1n, vol2n, vol3n, vol4n = vol(1,2,3,5)

---

一点乐趣（不推荐）：

• 让我们说你的vol1 =1, vol2 = 5 vol3 = 3 ...voln = 10
• 因此，让我们：

vol = [ 1, 5, 3, ... n values, 10]

• 现在，如果运行以下代码：

for j in range(len(vol)): exec(f'vol{j+1}n = [i/sum(vol) for i in vol][j]')

• 此代码将自动创建vol1n, vol2n, ... volnn变量

Answer 3

您可以检查是否使用“局部变量”函数定义了变量。或者，如果变量是全局定义的，则应使用“globals”函数

if 'vol1n' not in locals():
    vol1n = 0
else:
    vol1n=vol1/volsum

if 'vol2n' not in locals():
    vol2n = 0
else:
    vol2n=vol2/volsum

if 'vol3n' not in locals():
    vol3n = 0
else:
    vol3n=vol3/volsum

if 'vol4n'  not in locals():
    vol4n = 0
else:
    vol4n=vol4/volsum