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.

Parameters :

input : ndarray

Data from which to calculate center-of-mass.

labels : ndarray or None, optional

Labels for objects in input, as generated by ndimage.labels. Dimensions must be the same as input.

index : int, sequence of int, or None, optional

Labels for which to calculate centers-of-mass. If not specified, all labels greater than zero are used.

Returns :

centerofmass : tuple, or list of tuples

Co-ordinates of centers-of-masses.

Examples

>>> a = np.array(([0,0,0,0],
                  [0,1,1,0],
                  [0,1,1,0],
                  [0,1,1,0]))
>>> from scipy import ndimage
>>> ndimage.measurements.center_of_mass(a)
(2.0, 1.5)

Calculation of multiple objects in an image

>>> b = np.array(([0,1,1,0],
                  [0,1,0,0],
                  [0,0,0,0],
                  [0,0,1,1],
                  [0,0,1,1]))
>>> lbl = ndimage.label(b)[0]
>>> ndimage.measurements.center_of_mass(b, lbl, [1,2])
[(0.33333333333333331, 1.3333333333333333), (3.5, 2.5)]