As of a few days ago (Apache Spark 3.2 release), you can use the pandas API on a Spark cluster: https://databricks.com/blog/2021/10/04/pandas-api-on-upcomin...
You can keep using Panda code for exploratory work, then with a little effort move it to production with Spark runtime.