如何在不破坏循环的情况下返回值?

时间:2016-01-23 20:44:08

标签: python function return-value

我想知道如何在不破坏Python循环的情况下返回值。

以下是一个例子:

def myfunction():
    list = ['a', 'b', 'c', 'd', 'e', 'f', 'g']
    print(list)
    total = 0
    for i in list:
        if total < 6:
            return i  #get first element then it breaks
            total += 1
        else:
            break

myfunction()

return只会得到第一个答案,然后离开循环,我不想要,我想要返回多个元素,直到该循环结束。

如何解决这个问题,有什么解决方案吗?

4 个答案:

答案 0 :(得分:6)

您可以为此创建generator,这样您就可以yield生成器中的值(使用yield语句后,您的函数将成为生成器。)

请参阅以下主题,以便更好地了解如何使用它:

使用生成器的示例:

def simple_generator(n):
    i = 0
    while i < n:
        yield i
        i += 1

my_simple_gen = simple_generator(3) // Create a generator
first_return = my_simple_gen.next() // 0
second_return = my_simple_gen.next() // 1

此外,您可以在循环开始之前创建list并将append项目创建到该列表,然后返回该列表,这样该列表可以被视为结果列表&#34;返回&#34 ;在循环中。

使用list返回值的示例:

def get_results_list(n):
    results = []
    i = 0
    while i < n:
        results.append(i)
        i += 1
    return results


first_return, second_return, third_return = get_results_list(3)

注意:在带有列表的方法中,您必须知道函数将在results列表中返回多少值以避免too many values to unpack错误

答案 1 :(得分:5)

使用generator是可能的方法:

def myfunction():
    l = ['a', 'b', 'c', 'd', 'e', 'f', 'g']
    total = 0
    for i in l:
        if total < 6:
            yield i  #yields an item and saves function state
            total += 1
        else:
            break

g = myfunction()

现在,您可以通过调用yield i来访问通过next()返回的所有元素:

>>> val = next(g)
>>> print(v)
a
>>> v = next(g)
>>> print(v)
b

或者,在for循环中执行:

>>> for returned_val in myfunction():
...    print(returned_val)
a
b
c
d
e
f

答案 2 :(得分:2)

使用列表切片最容易表达您想要的内容:

socket.shutdownInput()

或者创建另一个列表,您将在函数末尾返回。

>>> l = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i']
>>> l[:6]
# ['a', 'b', 'c', 'd', 'e', 'f']

答案 3 :(得分:0)

创建生成器的yield语句就是您想要的。

What does the "yield" keyword do in Python?

然后使用下一个方法获取循环返回的每个值。

var = my_func().next()