From Pandas to Apache Spark’s Dataframe

With the introduction in Spark 1.4 of Window operations, you can finally port pretty much any relevant piece of Pandas’ Dataframe computation to Apache Spark parallel computation framework using Spark SQL’s Dataframe. If you’re not yet familiar with Spark’s Dataframe, don’t hesitate to checkout my last article RDDs are the new bytecode of Apache Spark and come back here after :p.

I figured some feedback on how to port existing « complex » code might be useful so the goal of this article will be to take a few concepts from Pandas Dataframe and see how we can translate this to PySpark’s Dataframe using Spark > 1.4.

Disclaimer:  a few operations that you can do in Pandas don’t have any sense using Spark. Please remember that Dataframes in Spark are like RDD in the sense that they’re an immutable data structure. Therefore things like :

df['three'] = df['one'] * df['two'] # to create a new column "three"

Can’t exist, just because this kind of affectation goes against the principles of Spark. Another example would be trying to access by index a single element within a Dataframe. Don’t forget that you’re using a distributed data structure, not an in-memory random-access data structure.

To be clear, this doesn’t mean that you can’t do the same kind of thing (i.e. create a new column) using Spark, it means that you have to think immutable/distributed and re-write parts of your code, mostly the parts that are not purely thought-of as transformations on a stream of data.

So let’s dive in.

Column selection

This part is not that much different in Pandas and Spark, but you have to take into account the immutable character of your dataframe. First let’s create two dataframes one in Pandas *pdf* and one in Spark *df* :

# Pandas => pdf
In [17]: pdf = pd.DataFrame.from_items([('A', [1, 2, 3]), ('B', [4, 5, 6])])

In [18]: pdf.A
0    1
1    2
2    3
Name: A, dtype: int64

# SPARK SQL => df
In [19]: df = sqlCtx.createDataFrame([(1, 4), (2, 5), (3, 6)], ["A", "B"])

In [20]: df
Out[20]: DataFrame[A: bigint, B: bigint]

In [21]:

Now in Spark SQL or Pandas you use the same syntax to refer to a column :

In [27]: df.A
Out[27]: Column<A>

In [28]: df['A']
Out[28]: Column<A>

In [29]: pdf.A
0    1
1    2
2    3
Name: A, dtype: int64

In [30]: pdf['A']
0    1
1    2
2    3
Name: A, dtype: int64

The output seems different, but these are still the same ways of referencing a column using Pandas or Spark, the only difference is that in Pandas, it is a mutable data structure that you can change, not in Spark.

Column adding

In [31]: pdf['C'] = 0

In [32]: pdf
   A  B  C
0  1  4  0
1  2  5  0
2  3  6  0

# In Spark SQL you'll use the withColumn or the select method,
# but you need to create a "Column", a simple int won't do :
In [33]: df.withColumn('C', 0)
AttributeError                            Traceback (most recent call last)
<ipython-input-33-fd1261f623cf> in <module>()
----> 1 df.withColumn('C', 0)

/Users/ogirardot/Downloads/spark-1.4.0-bin-hadoop2.4/python/pyspark/sql/dataframe.pyc in withColumn(self, colName, col)
   1196         """
-> 1197         return'*', col.alias(colName))
   1199     @ignore_unicode_prefix

AttributeError: 'int' object has no attribute 'alias'

# Here's your new best friend "pyspark.sql.functions.*"
# If you can't create it from composing columns
# this package contains all the functions you'll need :
In [35]: from pyspark.sql import functions as F
In [36]: df.withColumn('C', F.lit(0))
Out[36]: DataFrame[A: bigint, B: bigint, C: int]

In [37]: df.withColumn('C', F.lit(0)).show()

Most of the time in Spark SQL you can use Strings to reference columns but there are two cases where you’ll want to use the Column objects rather than Strings :

  • In Spark SQL Dataframe columns are allowed to have the same name, they’ll be given unique names inside of Spark SQL, but this means that you can’t reference them with the column name only as this becomes ambiguous.
  • When you need to manipulate columns using expressions like « Adding two columns to each other », « Twice the value of this column » or even « Is the column value larger than 0 ? », you won’t be able to use simple strings and will need the Column reference
  • Finally if you need renaming, cast or any other complex feature, you’ll need the Column reference too.

Here’s an example :

In [39]: df.withColumn('C', df.A * 2)
Out[39]: DataFrame[A: bigint, B: bigint, C: bigint]

In [40]: df.withColumn('C', df.A * 2).show()

In [41]: df.withColumn('C', df.B > 0).show()
|A|B|   C|

When you’re selecting columns, to create another *projected* dataframe, you can also use expressions :

In [42]: > 0)
Out[42]: DataFrame[(B > 0): boolean]

In [43]: > 0).show()
|(B > 0)|
|   true|
|   true|
|   true|

As you can see the column name will actually be computed according to the expression you defined, if you want to rename this, you’ll need to use the alias method on Column :

In [44]: > 0).alias("is_positive")).show()
|       true|
|       true|
|       true|

All of the expressions that we’re building here can be used for Filtering, Adding a new column or even inside Aggregations, so once you get a general idea of how it works, you’ll be fluent throughout all of the Dataframe manipulation framework.


Filtering is pretty much straightforward too, you can use the *RDD-like* filter method and copy any of your existing Pandas expression/predicate for filtering :

In [48]: pdf[(pdf.B > 0) & (pdf.A < 2)]
   A  B  C
0  1  4  0

In [49]: df.filter((df.B > 0) & (df.A < 2)).show()

In [55]: df[(df.B > 0) & (df.A < 2)].show()


What can be confusing at first in using aggregations is that the minute you write groupBy you’re not using a Dataframe object, you’re actually using a GroupedData object and you need to precise your aggregations to get back the output Dataframe :

In [77]: df.groupBy("A")
Out[77]: < at 0x10dd11d90>

In [78]: df.groupBy("A").avg("B")
Out[78]: DataFrame[A: bigint, AVG(B): double]

In [79]: df.groupBy("A").avg("B").show()
|1|   4.0|
|2|   5.0|
|3|   6.0|

As a syntactic sugar if you need only one aggregation, you can use the simplest functions like : avg, cout, max, min, mean and sum directly on GroupedData, but most of the time, this will be too simple and you’ll want to create a few aggregations during a single groupBy operation. After all (c.f. RDDs are the new bytecode of Apache Spark ) this is one of the greatest features of the Dataframes. To do so you’ll be using the agg method :

In [83]: df.groupBy("A").agg(F.avg("B"), F.min("B"), F.max("B")).show()
|1|   4.0|     4|     4|
|2|   5.0|     5|     5|
|3|   6.0|     6|     6|

Of course, just like before, you can use any expression especially column compositions, alias definitions etc… and some other non-trivial functions :

In [84]: df.groupBy("A").agg(
   ....: F.first("B").alias("my first"),
   ....: F.last("B").alias("my last"),
   ....: F.sum("B").alias("my everything")
   ....: ).show()
|A|my first|my last|my everything|
|1|       4|      4|            4|
|2|       5|      5|            5|
|3|       6|      6|            6|

Complex operations & Windows

Now that Spark 1.4 is out, the Dataframe API provides an efficient and easy to use Window-based framework – this single feature is what makes any Pandas to Spark migration actually do-able for 99% of the projects – even considering some of Pandas’ features that seemed *hard* to reproduce in a distributed environment.

A simple example that we can pick is that in Pandas you can compute a diff  on a column and Pandas will compare the values of one line to the last one and compute the difference between them. Typically the kind of feature hard to do in a distributed environment because each line is supposed to be treated independently, now with Spark 1.4 window operations you can define a window on which Spark will « execute some aggregation functions » but relatively to a specific line. Here’s how to port some existing Pandas code using diff :

In [86]: df = sqlCtx.createDataFrame([(1, 4), (1, 5), (2, 6), (2, 6), (3, 0)], ["A", "B"])

In [95]: pdf = df.toPandas()

In [96]: pdf
   A  B
0  1  4
1  1  5
2  2  6
3  2  6
4  3  0

In [98]: pdf['diff'] = pdf.B.diff()

In [102]: pdf
   A  B  diff
0  1  4   NaN
1  1  5     1
2  2  6     1
3  2  6     0
4  3  0    -6

In Pandas you can compute a diff on an arbitrary column, with no regard for keys, no regards for order or anything. It’s cool… but most of the time not exactly what you want and you might end up cleaning up the mess afterwards by setting the column value back to NaN from one line to another when the keys changed.

Here’s how you can do such a thing in PySpark using Window functions, a Key and, if you want, in a specific Order :

In [107]: from pyspark.sql.window import Window

In [108]: window_over_A = Window.partitionBy("A").orderBy("B")

In [109]: df.withColumn("diff", F.lead("B").over(window_over_A) - df.B).show()
|  A|  B|diff|
|  1|  4|   1|
|  1|  5|null|
|  2|  6|   0|
|  2|  6|null|
|  3|  0|null|

With that you are now able to compute a diff line by line – ordered or not – given a specific key. The great point about Window operation is that you’re not actually breaking the structure of your data. Let me explain myself.

When you’re computing some kind of aggregation (once again according to a key), you’ll usually be executing a groupBy operation given this key and compute the multiple metrics that you’ll need (if you’re lucky *at the same time*, if you’re not in multiple reduceByKey or aggregateByKey transformations).

But whether you’re using RDDs or Dataframe, if you’re not using window operations then you’ll actually crush your data in a part of your flow and then you’ll need to join back again the results of your aggregations to the *main*-dataflow. Window operations allows you to execute your computation and copy the results as additional columns without any explicit join.

This is a quick way to enrich your data adding rolling computations as just another column directly. Two additional resources are worth noting regarding these new features, the official Databricks blog article on Window operations and Christophe Bourguignat‘s article evaluating Pandas and Spark Dataframe differences.

To sum up you now have all the tools you need in Spark > 1.4 to port any Pandas computation in a distributed environment using the *very* similar Dataframe API.



Votre commentaire

Entrez vos coordonnées ci-dessous ou cliquez sur une icône pour vous connecter:


Vous commentez à l’aide de votre compte Déconnexion /  Changer )

Photo Facebook

Vous commentez à l’aide de votre compte Facebook. Déconnexion /  Changer )

Connexion à %s

%d blogueurs aiment cette page :