生成器函數Python 2中的返回值

Question

我想知道如何編寫一個生成器函數，該函數還可以選擇返回值。 在Python 2中，如果生成器函數嘗試返回值，則會收到以下錯誤消息。 SyntaxError: 'return' with argument inside generator

是否可以在我指定是否要接收生成器的地方編寫函數？

例如：

def f(generator=False):
    if generator:
        yield 3
    else:
        return 3

Answer 1

必讀： 了解Python中的生成器

關鍵信息：

函數中任何地方的yield使其成為生成器。

解析代碼時，功能被標記為生成器。 因此，不可能根據運行時傳遞的參數來切換函數行為（生成器/非生成器）。