[jira] [Created] (SPARK-17514) df.take(1) and df.limit(1).collect() perform differently in Python

["Josh Rosen (JIRA)" <ji...@apache.org>]

Josh Rosen created SPARK-17514:
----------------------------------

             Summary: df.take(1) and df.limit(1).collect() perform differently in Python
                 Key: SPARK-17514
                 URL: https://issues.apache.org/jira/browse/SPARK-17514
             Project: Spark
          Issue Type: Bug
          Components: PySpark, SQL
            Reporter: Josh Rosen
            Assignee: Josh Rosen


In PySpark, {{df.take(1)}} ends up running a single-stage job which computes only one partition of {{df}}, while {{df.limit(1).collect()}} ends up computing all partitions of {{df}} and runs a two-stage job. This difference in performance is confusing, so I think that we should generalize the fix from SPARK-10731 so that {{Dataset.collect()}} can be implemented efficiently in Python.



--
This message was sent by Atlassian JIRA
(v6.3.4#6332)

---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscribe@spark.apache.org
For additional commands, e-mail: issues-help@spark.apache.org