循环使用参数数组的自制函数,并在python中获取数组

时间:2017-01-02 20:52:07

标签: python python-3.x

有一个功能

def my_function (argument):
    summe=argument+5
    print(summe)
    return

我想循环使用不同参数的函数,比如

arguments=linspace(-0.1,-8.0,80)

将每个和(summe)一起保存在一个数组中

那么我怎么能用不同的参数重复一个函数(上面的简单函数只是一个例子)?

结果看起来像:

[4.9,4.8,4.7 ... -2.8,-2.9,-3.0]

1 个答案:

答案 0 :(得分:2)

您需要return来自您的函数的总和,如下所示:

def my_function (argument):
    summe = argument + 5
    print(summe)
    return summe

然后,假设您在数组中有一些参数,例如:

arguments = [-0.1,8.0,80]

您可以使用列表推导循环遍历这些参数并将函数调用的结果存储在另一个数组中:

result = [my_function(arg) for arg in arguments]