arrays - 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()，但更好的是从一开始就使用NumPy数组，无论在哪里你得到你的数据。