如何修改Python函数，使其循环调用时符合预期且不终止内核

给定以下函数：

def my_func(x):
    if x > 5:
        print(x)
    else:
        quit()

    print('this should be printed only if x > 5')

在循环中调用该函数：

for i in [2, 3, 4, 5, 6, 7]:
    my_func(i)

预期输出为：

6
this should be printed only if x > 5
7
this should be printed only if x > 5

但使用quit()会直接断开Kernel，导致循环提前终止。已知以下函数可实现需求，但不想将第二个print放在if分支内：

def my_func(x):
    if x > 5: 
        print(x)
        print('this should be printed only if x > 5')
    else:
        pass

另外，若将循环放入函数内可使用continue或break，但希望保持函数简洁，仅在循环中调用函数。请问需修改第一个函数的哪部分才能实现预期输出？

把else分支里的quit()替换成return即可。修改后的函数如下：

def my_func(x):
    if x > 5:
        print(x)
    else:
        return  # 结束当前函数调用，回到循环继续下一次迭代

    print('this should be printed only if x > 5')

原理说明 #

• quit()（或exit()）会直接终止整个Python解释器进程，一旦触发，后续的循环迭代和代码都不会执行，这就是导致Kernel断开的原因。
• return是函数的退出语句，只会结束当前这一次my_func的调用，程序会回到循环中继续执行下一次迭代，完全符合需求：当x<=5时跳过后续的print，且不影响循环继续运行。

内容的提问来源于stack exchange，提问作者Saeed