scipy.ndimage.measurements.center_of_mass

scipy.ndimage.measurements.center_of_mass(input, labels=None, index=None)

Calculate the center of mass of the values of an array at labels.

Labels must be None or an array of the same dimensions as the input.

Index must be None, a single label or sequence of labels. If none, all values where label is greater than zero are used.