Use take() instead of takeSample() in PySpark kmeans example.

This is a temporary change until we port takeSample().