[examples][mllib][python] SPARK-3838

This pull request refers to issue: https://issues.apache.org/jira/browse/SPARK-3838

Python example for word2vec