为什么函数通过打印和返回给出不同的结果？

Question

如果我编码：

a = [56, 2, 5, 2, 5, 2, 6, 2, 6, 2, 56, 2, 6, 2, 52, 5, 2]


def func1(a3):
    for i in range(len(a)-1):
        if a[i] + a[i+1] == 58:
            print(a[i])
func1(a)

我得到输出: 56 2 56

但是，如果我代码返回ai而不是print(ai)，我只能得到第一个合适的数字，我如何修复它呢？

Answer 1

当您的代码遇到返回语句时退出该函数。因此，以下迭代不会发生。如果您需要要打印的值列表，则接受一个空数组，然后在需要打印时追加到它，然后在for循环结束时返回该列表。

a = [56, 2, 5, 2, 5, 2, 6, 2, 6, 2, 56, 2, 6, 2, 52, 5, 2]


def func1(a3):
    ans = []
    for i in range(len(a)-1):
        if a[i] + a[i+1] == 58:
            ans.append(a[i])
    return ans
func1(a)

Answer 2

return函数使您的函数“停止运行”，返回传入参数的值。如果希望返回比第一个元素更多的元素，则可能希望创建一个列表，在该列表中添加所需的元素。

a = [56, 2, 5, 2, 5, 2, 6, 2, 6, 2, 56, 2, 6, 2, 52, 5, 2]
def func1(a3):
    result = []
    for i in range(len(a)-1):
        if a[i] + a[i+1] == 58:
            result += [a[i]]
func1(a)