NumPy 沿不相交索引求和

我有一个应用程序，需要对 3D NumPy 数组中的任意索引组进行求和。内置 NumPy 数组求和例程将沿 ndarray 维度之一的所有索引相加。相反，我需要沿着数组中的一个维度总结索引范围并返回一个新数组。

例如，假设我有一个形状为 ndarray(70,25,3)。我希望总结沿某些索引范围的第一个维度并返回一个新的 3D 数组。考虑总和0:25, 25:50 and 50:75这将返回一个形状数组(3,25,3).

有没有一种简单的方法可以沿着 NumPy 数组的一个维度进行“不相交和”来产生这个结果？

您可以使用np.add.reduceat作为解决这个问题的一般方法。即使范围的长度不一样，这也有效。

对切​​片求和0:25, 25:50 and 50:75沿轴 0，传入索引[0, 25, 50]:

np.add.reduceat(a, [0, 25, 50], axis=0)

此方法还可用于对非连续范围求和。例如，对切片求和0:25, 37:47 and 51:75, write:

np.add.reduceat(a, [0,25, 37,47, 51], axis=0)[::2]

求和范围的另一种方法samelength 是重塑数组，然后沿轴求和。与上面第一个示例等效的是：

a.reshape(3, a.shape[0]//3, a.shape[1], a.shape[2]).sum(axis=1)