ENH Add return_* arguments to kmeans()

This is especially relevant for use with jug. This way, if only
interested in one of the outputs, the return value is a single numpy
array, which jug handles particularly well.