锥形积分

Question

是否有一种快速的方法（标准方法或通过某些库）来计算python中圆锥上的积分？

即我有一个包含一些值的数组A，并且我想在由数组sliceNums中的数字指定的索引锥上进行计算和积分。我可以直接做到：

n = 10; m=100                         # just as example
A = np.arange(n*m).reshape( (n,m) )   # just as example
sliceNums = np.arange(n)*2            # just as example
B = np.zeros(n)
for i in range(n):
    B[i] = np.sum ( A[i,sliceNums[i]: ] )

但是对于大的n和m来说，这会很慢。有没有一种更快的计算B的方法？ （sliceNums元素可以是0到m之间的任何数字）