返回结果而不是在函数中打印

Question

我有一个关于 Python 的问题。

我有一个递归函数：

def func(num, k):

    if k == 0:
        print(num, end = '')
        return

    for item in num:
        if item == '1':
            func('123', k-1)
        elif item == '2':
            func('321', k-1)
        elif item == '3':
            func('1', k-1)

运行此代码func('2', 3) 打印： 12313211231233211

但是，实际上我不希望该函数打印结果。我想把它退回来。所以，如果我以a = func('2', 3) 运行程序，那么程序什么也不会打印。现在我可以在函数外使用a，例如打印它print(a)，得到12313211231233211。

我应该如何更改返回结果而不打印的代码？

谢谢！

Answer 1

这是一种无需更改函数代码的方法。相反，我们将创建一个函数装饰器，它会静默捕获所有打印输出，并将其保存以供以后使用。

import sys
from io import StringIO

def returnStdout(fn):
    silent_stdout = StringIO()
    def silentFn(*args, depth=0, **kargs):
        nonlocal silent_stdout

        #only replace stdout if it hasn't been replaced already.
        #this may be important if we decorate a recursive function.
        if sys.stdout != silent_stdout:
            old_stdout = sys.stdout
            sys.stdout = silent_stdout
            fn(*args, **kargs)
            sys.stdout = old_stdout
            result = silent_stdout.getvalue()

            #clear silent_stdout so the next call to the decorated function doesn't contain the values of previous calls
            silent_stdout = StringIO()

            return result
        else:
            fn(*args, **kargs)
    return silentFn

@returnStdout
def func(num, k):

    if k == 0:
        print(num, end = '')
        return

    for item in num:
        if item == '1':
            func('123', k-1)
        elif item == '2':
            func('321', k-1)
        elif item == '3':
            func('1', k-1)


result = func("2", 3)
print("got result. Result is:")
print(result)

结果：

got result. Result is:
12313211231233211

Answer 2

def func(num, k, result=None):
    if result is None:
        result = []
    if k == 0:
        result.append(num)
    else:
        for item in num:
            if item == '1':
                func('123', k-1, result)
            elif item == '2':
                func('321', k-1, result)
            elif item == '3':
                func('1', k-1, result)
    return "".join(result)

我添加了result 参数，但只是为了保存解决方案，您不需要提供任何列表来运行此版本。就像你发布的那样运行它：

print func('2', 3)

Answer 3

这是另一种方法。

def func(num, k, m={'1':'123','2':'321','3':'1'}):
    def recurs(s, c, r=''):
        return s if c == 0 else ''.join([recurs(m.get(i,''), c-1) for i in s])
    return recurs(num, k)

>>> func('2', 3)
'12313211231233211'

Answer 4

在 Python 3.3 中：

def iter_func(num, k):
    if k == 0:
        yield num
    else:
        for item in num:
            if item == '1':
                yield from iter_func('123', k-1)
            elif item == '2':
                yield from iter_func('321', k-1)
            else:
                yield from iter_func('1', k-1)

def func(num, k):
    return ''.join(iter_func(num, k))

在早期的 Python 版本中没有yield from，但您可以通过以下方式实现相同的目的：

for x in whatever:
    yield x

顺便说一句，你也可以通过重组来简化一点，特别是如果你不使用 3.3，因此在每种情况下都需要额外的代码：

subvalues = {'1' : '123', '2': '321'}

...

for item in num:
    yield from iter_func(subvalues.get(item, '1'), k-1)

同样的事情的另一种说法：

import itertools

subvalues = {'1' : '123', '2': '321'}
def iter_func(num, k):
    if k == 0:
        return (num,)
    return itertools.chain.from_iterable(
        iter_func(subvalues.get(item, '1'), k-1)
        for item in num
    )