Python在二进制numpy数组中找到1d质心的最佳方法

Question

假设我有下面的Numpy数组，其中我只有一个连续的1s切片：

import numpy as np
x = np.array([0,0,0,0,1,1,1,0,0,0], dtype=1)

我想找到1元素的一维质心指数。我可以键入以下内容：

^{pr2}$

（当然，当1s切片的元素数为偶数时，这将导致粗略近似。） 有没有更好的方法可以做到这一点，比如计算效率更高的一行程序？在

Answer 1

你不能简单地做下面的事吗？在

center_of_mass = (x*np.arange(len(x))).sum()/x.sum() # 5

%timeit center_of_mass = (x*arange(len(x))).sum()/x.sum()
# 100000 loops, best of 3: 10.4 µs per loop