SparseRespUtil.py 文件源码

 复制代码def test_speed_np_builtins(size, nLoop, nRep=1):
    setupCode = (
        "import numpy as np;" +
        "PRNG = np.random.RandomState(0);" +
        "x = PRNG.rand(%d);" % (size)
    )
    pprint_timeit(
        stmt='np.argmax(x)',
        setup=setupCode, number=nLoop, repeat=nRep)

    pprint_timeit(
        stmt='np.argsort(x)',
        setup=setupCode, number=nLoop, repeat=nRep)

    nnzPerRows = [0]
    for expval in np.arange(0, np.ceil(np.log2(size / 2))):
        nnzPerRows.append(2**expval)

    for nnzPerRow in nnzPerRows:
        funcCode = 'np.argpartition(x, %d)' % (nnzPerRow)
        pprint_timeit(
            stmt=funcCode, setup=setupCode, number=nLoop, repeat=nRep)