Databricks Use Python Variable Assignment