如何使用Python timeit模块捕获返回值？ - python

我在for循环中使用sklearn运行了几种机器学习算法，并希望了解每种算法花费的时间。问题是我还需要返回一个值，并且DONT希望多次运行它，因为每种算法都花很长时间。有没有一种方法可以使用python的timeit模块或具有类似功能的类似模块捕获返回值'clf'...

def RandomForest(train_input, train_output):
    clf = ensemble.RandomForestClassifier(n_estimators=10)
    clf.fit(train_input, train_output)
    return clf

当我这样调用函数时

t = Timer(lambda : RandomForest(trainX,trainy))
print t.timeit(number=1)

附言我也不想设置全局的“clf”，因为以后可能要进行多线程或多处理。

参考方案

问题归结为timeit._template_func不返回函数的返回值:

def _template_func(setup, func):
    """Create a timer function. Used if the "statement" is a callable."""
    def inner(_it, _timer, _func=func):
        setup()
        _t0 = _timer()
        for _i in _it:
            _func()
        _t1 = _timer()
        return _t1 - _t0
    return inner

我们可以通过一些猴子修补将timeit弯曲到我们的意愿:

import timeit
import time

def _template_func(setup, func):
    """Create a timer function. Used if the "statement" is a callable."""
    def inner(_it, _timer, _func=func):
        setup()
        _t0 = _timer()
        for _i in _it:
            retval = _func()
        _t1 = _timer()
        return _t1 - _t0, retval
    return inner

timeit._template_func = _template_func

def foo():
    time.sleep(1)
    return 42

t = timeit.Timer(foo)
print(t.timeit(number=1))

退货

(1.0010340213775635, 42)

第一个值是时间结果(以秒为单位)，第二个值是函数的返回值。

请注意，当将可调用对象传递给timeit时，上述的猴子补丁只会影响timeit.Timer的行为。如果传递字符串语句，则必须(类似)猴子修补timeit.template字符串。

Python uuid4，如何限制唯一字符的长度 - python

在Python中，我正在使用uuid4（）方法创建唯一的字符集。但是我找不到将其限制为10或8个字符的方法。有什么办法吗？uuid4()ffc69c1b-9d87-4c19-8dac-c09ca857e3fc谢谢。 参考方案 尝试：x = uuid4() str(x)[:8] 输出："ffc69c1b" Is there a way to…

Python:无法识别Pip命令 - python

这是我拍摄的屏幕截图。当我尝试在命令提示符下使用pip时，出现以下错误消息:pip无法识别为内部或外部命令，可操作程序或批处理文件。我已经检查了这个线程:How do I install pip on Windows?我所能找到的就是我必须将"C:\PythonX\Scripts"添加到我的类路径中，其中X代表python版本。如您在我的…

Python:如何将有效的uuid从String转换为UUID？ - python

我收到的数据是 { "name": "Unknown", "parent": "Uncategorized", "uuid": "06335e84-2872-4914-8c5d-3ed07d2a2f16" }, 我需要将uuid从Strin…

Python 3会流行吗？ - python

我已经学习了一些Python 2和Python 3，似乎Python 2总体上比Python 3更好。这就是我的问题所在。是否有充分的理由真正切换到python 3？ 参考方案 总体上，甚至在大多数细节上，Python3都比Python2更好。关于第三方库， Python 3落后于的唯一区域是。使Python变得如此出色的原因不仅在于它作为一种语言的内在特性…

Python-如何检查Redis服务器是否可用 - python

我正在开发用于访问Redis Server的Python服务(类)。我想知道如何检查Redis Server是否正在运行。而且如果某种原因我无法连接到它。这是我的代码的一部分import redis rs = redis.Redis("localhost") print rs 它打印以下内容<redis.client.Redis o…