[SPARK-2871] [PySpark] Add missing API

Try to bring all Java/Scala API to PySpark.
If you'd like you can implement lookup() the same way as in Scala, it's not too bad