Pyspark Check If Column Is Array Type