Created
September 17, 2016 09:00
-
-
Save pavlov99/5f03e6cdee660f69235cff44b68b164d to your computer and use it in GitHub Desktop.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import org.apache.spark.sql.Window | |
val competitorWindow = Window | |
.partitionBy("date", "competitor") | |
.orderBy(levenshtein($"competitor", $"short_name")) | |
val scheduleRich = schedule | |
.join( | |
teams, levenshtein($"competitor", $"short_name") < 5, "left_outer" | |
) | |
.withColumn("_rank", row_number().over(competitorWindow)) | |
.filter($"_rank" === 1) | |
.drop("_rank") | |
scheduleRich.drop("short_name").show(2) | |
+----------+----------+---------------+--------+------------------+ | |
| date|competitor| team|division| conference| | |
+----------+----------+---------------+--------+------------------+ | |
|2016-10-05| Anaheim| Anaheim Ducks| Pacific|Western Conference| | |
|2016-10-09| Anaheim| Anaheim Ducks| Pacific|Western Conference| | |
+----------+----------+---------------+--------+------------------+ |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment