如何在Python中将变量传入和传出函数[重复]

Question

当我在 Python 中编写函数时，我通常需要向函数传递相当多的变量。此外，此类函数的输出包含多个变量。为了管理这些变量 I/O，我使用字典数据类型，将所有输入变量打包到字典中以注入函数，然后在函数末尾编译另一个字典以返回主程序。这当然需要对输出字典进行另一次解包。

dict_in = {'var1':var1,
           'var2':var2,
           'varn':varn}    

def foo(dict_in):
    var1 = dict_in['var1']
    var2 = dict_in['var2']
    varn = dict_in['varn']

    """ my code """

    dict_out = {'op1':op1,
                'op2':op2,
                'op_m':op_m}
    return dict_out

随着变量列表的增加，我怀疑这将是一种处理变量 I/O 的低效方法。

有人可以建议一种更好、更有效且不易出错的方法吗？

Answer 1

您可以利用 kwargs 来解压命名变量

def foo(**kwargs):
    kwargs['var1'] = do_something(kwargs['var1'])
    ...
    return kwargs

Answer 2

如果您发现自己编写了许多作用于相同数据的函数，一种更好的方法是使用类来包含您的数据。

class Thing:
    def __init__(self, a, b, c):
        var_1 = a
        var_2 = b
        var_3 = c

    # you can then define methods on it

    def foo(self):
        self.var_1 *= self.var_2

# and use it
t = Thing(1, 2, 3)
t.foo()
print(t.var_1)

有许多方法可以更简单地创建这些。其中一些包括：

attrs:

>>> @attr.s
... class SomeClass(object):
...     a_number = attr.ib(default=42)
...     list_of_numbers = attr.ib(default=attr.Factory(list))
...
...     def hard_math(self, another_number):
...         return self.a_number + sum(self.list_of_numbers) * another_number

namedtuples

>>> Point = namedtuple('Point', ['x', 'y'])
>>> p = Point(11, y=22)     # instantiate with positional or keyword arguments
>>> p.x + p.y               # fields accessible by name
33

数据类

这些还没有在 python 中，而是will be added in 3.7。我在这里添加它们是因为它们可能会成为未来的首选工具。