加速求和运算的索引

使用numpy中的sum函数代替for循环进行求和操作。

示例代码如下：

import numpy as np

# 创建一个长度为1000的一维数组
arr = np.arange(1000)

# 使用for循环进行求和
sum1 = 0
for i in range(1000):
    sum1 += arr[i]

# 使用numpy的sum函数进行求和
sum2 = np.sum(arr)

assert sum1 == sum2 # 验证两种方法得到的结果相同

使用numpy中的sum函数可以大大加快求和操作的速度，特别是在处理大量数据时更加明显。