Work with DataFrames columns in Azure Databricks