Pyspark Select Row Shortcut