Python增加数组的一部分

时间:2014-10-17 06:52:33

标签: arrays python-2.7 numpy

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

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

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

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

谢谢,

马克

1 个答案:

答案 0 :(得分:1)

NumPy确实是一个解决方案:

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

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