如何使用函数的返回值作为python中返回元组的条件

Question

我一直在找这样的东西，但我找不到，所以就来了。

一些背景

我使用opencv从视频文件中检索帧。通常人们会无休止地循环这样做：

while (True):
    s, img = cv.read()

或

for i in xrange(10000): #just a big number
    s, img = cv.read()

现在我想检索所有帧并在没有更多帧时退出循环。然而，我在python中的技能还不够强，无法做我想做的事情。

我想知道的是

read函数（或方法，我不知道它们在python中是如何调用的）返回一个元组：首先代表操作成功，其次代表返回的帧。当元组的第一个元素为false时，我想打破while循环。有C背景，我想也许这可以工作：

while ((success, img = capture.read())[0]):
    #do sth with img

我认为这将打破成功是假的循环。但事实并非如此。然后我想也许这会奏效：

while ((success, img = capture.read()).success):
    #do sth with img

它也不起作用。我不想做这样的事情

while(True):
    s, i = capture.read()
    if (s == False):
        break

如何在而中测试条件，而不是在if中如果成功则中断？

Answer 1

Python有一个可选的iter签名，它将函数作为第一个参数，将确定停止条件的哨兵作为第二个参数。

使用它，您可以得出如下结果：

for s,img in iter(cv.read, (False, None)):
    print img

尽管如此，我怀疑它是否比在if块中简单地中断好。

此外，它只接受哨兵作为整体返回值，并且不能基于它的某些部分（例如元组的第一个值）的停止条件。这可以是解决方法，但会使代码更加模糊：

>>> for s,img in itertools.takewhile(lambda x: x[0], iter(cv.read, None)):
    print img

它使用itertools。使用while来确定返回元组的第一个值何时等于False。

测试的完整版本：

>>> class Capture(object):
    def __init__(self):
        self.L = iter([1,2,3,4,5])
    def read(self):
        try:
            img = self.L.next()
        except StopIteration:
            return (False,None)
        return True, img

>>> cv = Capture()
>>> for s,img in iter(cv.read, (False, None)):
    print img

1
2
3
4
5

>>> cv = Capture()
>>> for s,img in itertools.takewhile(lambda x: x[0], iter(cv.read, None)):
    print img


1
2
3
4
5

Answer 2

思考蟒蛇语的最好方法是忘记其他语言

s = True
while s:
    s, i = capture.read()
    if s:
        do_some_stuff(i)

Answer 3

您可以编写一个生成器函数。

def readframes(cv):
    while True:
        success, frame = cv.read()
        if success:
            yield frame
        else:
            return

这样，就可以使用for循环遍历帧。

for frame in readframes(cv):
    do_something_with_frame(frame)