Pyspark substring variable