Python增加数组的一部分

Question

我想尽快在python中增加数组的一部分。我使用一个简单的循环：

>>> test = [0,0,0,0,0]
>>> for i in xrange(1, 3):
    test[i] += 1

>>> test
[0,1,1,0,0]

在我的程序中，测试列表包含数百万个元素。也许numpy可能是解决方案？

谢谢，

马克

Answer 1

NumPy确实是一个解决方案：

import numpy as np
arr = np.array(test)
arr[1:3] += 1

如果你真的需要回到arr.tolist()，可以使用list但最好是从一开始就使用NumPy数组，无论你在哪里获得数据。