Question

这只是一个满足我好奇心的问题我实际上并没有计划将列表用作numba函数的参数。

但我想知道为什么将列表传递给numba函数似乎是O(n)操作，而在纯Python函数中它是O(1)操作。

一些简单的示例代码：

import numba as nb

@nb.njit
def take_list(lst):
    return None

take_list([1, 2, 3])  # warmup

时间安排：

for size in [10, 100, 1000, 10000, 100000, 1000000]:

    lst = [0]*size
    print(len(lst))
    %timeit take_list(lst)   # IPythons "magic" timeit

结果：

10
4.06 µs ± 26.1 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)
100
14 µs ± 360 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)
1000
109 µs ± 434 ns per loop (mean ± std. dev. of 7 runs, 10000 loops each)
10000
1.08 ms ± 17.8 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)
100000
10.7 ms ± 26.2 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)
1000000
112 ms ± 383 µs per loop (mean ± std. dev. of 7 runs, 10 loops each)

Answer 1

操作Python列表需要Python API调用，这在nopython模式下是禁止的。 Numba实际上将列表内容复制到自己的数据结构中，这需要时间与列表的大小成比例。

为什么将一个列表（长度为n）传递给numba nopython函数进行O（n）操作

1 个答案: