如果超过时间限制，Python跳过循环行并打印跳过的行

Question

类似的问题，但对于这个用例有点不同。我有一个在多个数据帧上运行的函数，我希望在每个数据帧运行时使用一个计时器，如果需要超过 30 秒的时间来跳过它并打印数据帧名称。

功能-

def some_function():
# runs on pd.DataFrames
    cols = [some code]
    rows = [some code]
    df = pd.DataFrame(rows, columns=cols)
    return df

df_1 = some_function()
df_2 = some_function()
df_3 = some_function()

其中df_1 和df_2 各花费不到30 秒，但df_3 花费3 分钟。然后它将运行df_1 和df_2，但打印出df_3 花费了超过30 秒。如何创建这样的定时循环？

编辑

在字典中组织数据帧的循环-

def some_function():
# runs on pd.DataFrames
    cols = [some code]
    rows = [some code]
    t1 = datetime.now()
    if (datetime.now()-t1).seconds <=30:
        df = pd.DataFrame(rows, columns=cols)
    return df

d = {}
dfs = range(0,10)
for name in dfs:
    d[name] = some_function()

我认为这可能会更容易，如果循环并花费太长时间然后跳过？

Answer 1

基于已编辑的问题。这里的想法是在您完成长时间工作的每个点检查超时条件。

def some_function():
# runs on pd.DataFrames
    t1 = datetime.now()
    cols = [some code]
    if (datetime.now()-t1).seconds > 30:
        return None
    rows = [some code]
    if (datetime.now()-t1).seconds > 30:
        return None
    return pd.DataFrame(rows, columns=cols)

Answer 2

您必须在 some_function() 中添加代码才能执行此操作。考虑，例如：

def some_function():
    timeout = time.time() + 30
    for some_very_large_loop:
        if time.time() > timeout:
            print( "timeout" )
            return None

如果您有多个处理循环，只需在每个循环中添加该子句。外部代码必须检查 None 并打印诊断信息。

Answer 3

为了停止执行，您需要使用线程。如果您只想表示执行时间过长，您可以编写一个装饰器来自动打印警告消息：

from time import time
def timeLimit(maxTime,name="function"):
    def decorated(f):
        def execute(*arg,**kwarg):
            start    = time()
            result   = f(*arg,**kwarg)
            duration = time()-start
            if duration>maxTime:
                sarg = ",".join(map(str,arg))
                print(f"{name}({sarg}) took more than {maxTime} seconds: {duration}")
            return result
        return execute
    return decorated

用法：

@timeLimit(2,"myfunction") # max 2 seconds
def myfunction(x):
    for _ in range(x*1000000): pass
    return x*x

result = myfunction(30)
result = myfunction(190)

'myfunction(190) took more than 2 seconds: 3.5628979206085205'