Skip to content

[SPARK-1308] [PySpark] Add partitions() method to PySpark RDDs #218

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
nchammas wants to merge 4 commits into from
Closed

[SPARK-1308] [PySpark] Add partitions() method to PySpark RDDs #218

Changes from 1 commit
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
e972d3e
[SPARK-1308] Add partitions() method to PySpark RDDs
nchammas Mar 24, 2014
d8682ec
Merge remote-tracking branch 'upstream/master'
nchammas Apr 3, 2014
88fa762
Merge remote-tracking branch 'upstream/master'
nchammas Apr 22, 2014
ab4467f
Changed partitions() to getNumPartitions()
nchammas Apr 22, 2014
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions python/pyspark/rdd.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -202,6 +202,12 @@ def flatMap(self, f, preservesPartitioning=False):
def func(s, iterator): return chain.from_iterable(imap(f, iterator))
return self.mapPartitionsWithIndex(func, preservesPartitioning)

def partitions(self):
"""
Get the array of partitions of this RDD, taking into account whether the RDD is checkpointed or not.
"""
return self._jrdd.splits()
Copy link
Contributor Author

@nchammas nchammas Mar 25, 2014

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actually, I'm not sure that having a Python method return a Java object is right thing to do. Or is it?

len(rdd.partitions()) works as you would expect, and so does rdd.partitions().size(), but I'm not sure how else this method might be used.


def mapPartitions(self, f, preservesPartitioning=False):
"""
Return a new RDD by applying a function to each partition of this RDD.
Expand Down