小编K. *_*hea的帖子

我有一个 0 和 1 的 numpy 数组（512 x 512）。我想计算1 形状的质心（它们都连接在阵列中间的一个圆形斑点中）。

for i in xrange(len(array[:,0])):
    for j in xrange(len(array[0,:])):
        if array[i,j] == 1:
            x_center += i
            y_center += j

count = (aorta == 1).sum()
x_center /= count
y_center /= count

有没有办法加快我上面的计算？我可以使用 numpy.where() 之类的吗？是否有任何 python 函数可以并行执行此操作？