如何在收益中扩展元组?

How to expand a tuple in a yield?

我试图在 yield 语句中扩展一个元组,但得到的是 SyntaxError。这甚至可能,还是 * 运算符仅用于函数调用?

下面是一个例子:

class Inner(object):
    def __init__(self):
        self._vals = {'a': 1,
                      'b': 2,
                      'c': 3,
                      'd': 4}

    def __iter__(self):
        for key, val in self._vals.items():
            yield key, val

class Outer(object):
    def __init__(self):
        self._morevals = {"foo": Inner(),
                          "bar": Inner()}

    def __iter__(self):
        for key, moreval in self._morevals.items():
            for val in moreval:
                yield key, *val    # !!! <--- This is where the error is

object_under_test = Outer()

for outer_key, inner_key, inner_value in object_under_test:
    print("{} {} {}".format(outer_key, inner_key, inner_value))

    # Want:
    # foo a 1
    # foo b 2
    # foo c 3
    # foo d 4
    # bar a 1
    # bar b 2
    # bar c 3
    # bar d 4

相反,我收到了这个错误:

    yield key, *val
               ^
SyntaxError: invalid syntax

在元组周围添加括号:

yield (key, *val)

yield 在这里很特别,因为表单只接受表达式列表,而不是 starred_list form.

您需要将其括起来。简而言之,使用:

yield (key, *val)

或者,您可以通过使用 yield from 理解来缩短它。总而言之,你的 __iter__ 看起来像:

 def __iter__(self):
    items = self._morevals.items()
    yield from ((k, *v) for k, vals in items for v in vals)

同样,在ClassOne.__iter__中使用yield from

def __iter__(self):
    yield from self._vals.items()