Numpy:如何以向量化的方式从另一个数组中获取基于索引的整数数组切片的总和？

给定两个整数数组a和b，其中b中的元素表示a的索引...

a = array([10,10,10,8,8,8])

b = array([0,2,3,5])

我想生成一个新的数组，它的元素是a中沿着b给出的索引范围的元素的和，不包括范围尾部的元素……这很难用语言表达，但上面给出的a和b的预期结果是：

result = array([0, # sum(a[:0])
                20,  # sum(a[0:2])
                10,  # sum(a[2:3])
                16])  # sum(a[3:5])

我如何才能以一种矢量化/“数字方式”的方式实现这一点？

谢谢!

转载请注明出处:http://www.intsu.net/article/20230518/2360277.html