Interactive data analysis with pyspark