Pyspark Check If Array Or String