是否有可能像回归一样产生两件事?

时间:2015-01-29 01:22:41

标签: python yield

def foo(choice):
    for i in limit:
        d1 = doSomeCalc()
        d2 = doSomeOtherCalc()
        if choice == "stuff":
            yield {
                d1 : "value"
            }
        else:
            yield {
                d2 : "Othervalue"
            }

我有一个yield两种类型字典的功能,具体取决于用户的选择

def bar():
    for i in limit:
        d1 = doSomeCalc()
        d2 = doSomeOtherCalc()
        return {d1 : "value"}, {d2 : "Othervalue"}

a,b = bar() // when function returns two dictionaries

就像return一样,我可以使用yield一次提供两个不同的词典吗?我如何获得每个值?

我现在不想将if-else保留在我的功能中。

2 个答案:

答案 0 :(得分:8)

您一次只能生成一个值。迭代生成器将依次产生每个值。

def foo():
  yield 1
  yield 2

for i in foo():
  print i

与往常一样,该值可以是一个元组。

def foo():
  yield 1, 2

for i in foo():
  print i

答案 1 :(得分:0)

另一种方法是产生像字典这样的数据结构,如下所示:

def create_acct_and_tbl():
    yield {'acct_id': 4, 'tbl_name': 'new_table_name'}


def test_acct_can_access():
    rslt = create_acct_and_tbl
    print(str(rslt['acct_id']), rslt['tbl_name'])