Find the centroid of a 2D array.

centroid.py

import numpy as np

print "\n\n"

# Width and height of the image.
W = 5
H = 5
# Generate a random image.
x = np.random.rand(W, H) > 0.5

# Print the image.
print x
print

# Print the iamge as bit cells.
print (x == 1).astype(int)
print

# Print the centroid.
xis, yis = np.nonzero(x)
print xis.mean(), yis.mean()
print

# print np.transpose(np.nonzero(x))
# print np.array(np.nonzero(x))[:,2]

?

np.mean(x.astype(int), axis=0)