在Python中将生成器表达式作为输入参数

时间:2017-02-12 04:00:15

标签: python-2.7 function input generator-expression

如何与生成器表达式的值进行交互?例如:

def sumValues(*args):
# compute the sum of values

sumValues(abs(x) for x in range(0,10))

当范围已知时(例如,我们知道范围是[0,10)),我们可以设置一个for循环,但如果范围未知会怎么样?

2 个答案:

答案 0 :(得分:1)

它取决于你希望如何调用你的函数,如果你想用list,generator或更一般的可迭代(可以在for循环中使用的任何东西)调用它并模仿构建的行为-in sum然后

def my_sum(iterable):
    total = 0
    for x in iterable:
        total += x
    return total

并使用

my_sum(range(10))
my_sum(abs(x) for x in range(0,10))
my_sum([1,2,3,4,5,6])

如果你想模仿max的行为并按上述方式使用它,也喜欢my_sum(1,2)my_sum(1,2,3)my_sum(1,2,3,4,5,6,7)等,那么

def my_sum(*argv):
    total = 0
    values = None
    if len(argv) == 1:
        values = argv[0]
    else:
        values = argv
    for x in values:
        total += x
    return total

答案 1 :(得分:0)

我想。这是答案:

def sumValues(*args):
    # compute the sum of values
    total = 0
    for item in args[0]:
        total += item
    return total

print(sumValues(abs(x) for x in range(0,10)))
[output] => 45