I would like to keep only the employees which does have a departement ID referenced in the second table.
Employee table
LastName DepartmentID
Rafferty 31
Jones 33
Heisenberg 33
Robinson 34
Smith 34
Department table
DepartmentID
31
33
I have tried the following code which does not work:
employee = [['Raffery',31], ['Jones',33], ['Heisenberg',33], ['Robinson',34], ['Smith',34]]
department = [31,33]
employee = sc.parallelize(employee)
department = sc.parallelize(department)
employee.filter(lambda e: e[1] in department).collect()
Py4JError: An error occurred while calling o344.__getnewargs__. Trace:
py4j.Py4JException: Method __getnewargs__([]) does not exist
Any ideas? I am using Spark 1.1.0 with Python. However, I would accept a Scala or Python answer.
Best Answer
In this case, what you would like to achieve is to filter at each partition with the data contained in the department table: This would be the basic solution:
If your department data is large, a broadcast variable will improve performance by delivering the data once to all the nodes instead of having to serialize it with each task
Although using join would work, it's a very expensive solution as it will require a distributed shuffle of the data (byKey) to achieve the join. Given that the requirement is a simple filter, sending the data to each partition (as shown above) will provide much better performance.