SPARK-1162 added takeOrdered to pyspark
Code review comments.