For Databricks Runtime 5.5 LTS, Spark jobs, Python notebook cells, and library installation all support both Python 2 and 3.
The default Python version for clusters created using the UI is Python 3. In Databricks Runtime 5.5 LTS the default version for clusters created using the REST API is Python 2.