python - 如何预分配列表列表？

Question

我正在使用以下代码创建列表列表:

zeroArray = [0]*Np
zeroMatrix = []
for i in range(Np):
    zeroMatrix.append(zeroArray[:])

有没有更有效的方法来做到这一点？我希望得到类似 zeroArray = [0]*Np; 的东西。 zeroMat = zeroArray*Np 但找不到类似的东西。

Answer 1

也许你应该考虑使用 NumPy .看起来你正在做数字工作，这就是它的目的。这是迄今为止最快的，不包括导入语句:

import numpy
Np = 80
zeroMatrix = numpy.zeros((Np, Np))

时间:

>python -m timeit -s "import numpy; Np = 80" "zeroMatrix = numpy.zeros((Np, Np))"
100000 loops, best of 3: 4.36 usec per loop

>python -m timeit -s "Np = 80" "zeroArray = [0]*Np" "zeroMatrix = [None] * Np" "for i in range(Np):" "  zeroMatrix[i] = zeroArray[:]"
10000 loops, best of 3: 62.5 usec per loop

>python -m timeit -s "Np = 80" "zeroMatrix = [[0] * Np for i in range (Np)]"
10000 loops, best of 3: 77.5 usec per loop

>python -m timeit -s "Np = 80" "zeroMatrix = [[0 for _ in range(Np)] for _ in range(Np)]"
1000 loops, best of 3: 474 usec per loop