๐Ÿ”— Referring to Columns in PySpark