【问题标题】:How do you pass arguments to a function within an array?如何将参数传递给数组中的函数?
【发布时间】:2019-05-16 19:44:33
【问题描述】:

假设我有一个函数数组,每个函数都有相同的参数,但每个函数都以不同的方式处理参数。然后我想遍历一个循环,一遍又一遍地使用我的函数数组来获取更新的值。如何将参数传递给函数?

def f_1(a, b, c):
    val = a + b + c
    return val

def f_2(a, b, c):
    val = a*b + c
    return val

# something like this?

fun_arr = [f_1, f_2]

val1 = [fun_arr[0](a1, b1, c1), fun_arr[1](a1, b1, c1)]
val2 = [fun_arr[0](a2, b2, c2), fun_arr[1](a2, b2, c2)]

我希望上面的伪代码有意义。这是我正在尝试做的简化版本。

如果上下文有帮助,我正在尝试为一个方程组编写一个 RK2 算法,该方程组可能会被重用于一组通用 ODE。我想我可以很容易地编写一些适用于我的特定问题的简单代码,但我想挑战自己以使我的代码可重用。

【问题讨论】:

标签: python numerical-methods


【解决方案1】:

像这样:

fun_arr = [f_1, f_2]

args_list = [
    (a1, b1, c1),
    (a2, b2, c2),
]

val1 = [f(*args_list[0]) for f in fun_arr]

all_vals = [
    [f(*args) for f in fun_arr]
    for args in args_list
    ]

【讨论】:

    【解决方案2】:
    def solver(fun_arr,*vals):
        new_vals = []
        for v in vals:
            for f in fun_arr:
                new_vals.append(f(v))
    

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 2011-09-24
      • 2013-09-27
      • 2015-10-30
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多