Manipulating your RDD in Python