title: Visualise grouped line plots in pyspark
body: <p>I have this <code>DF</code>(sample) and I am using PySpark in Databricks. I would like to have line plot <code>DATE</code> vs <code>BALANCE</code> but for each <code>ID</code> in one single frame.</p>
  <pre><code>+-------------+-------------+----------------+
  |         DATE|      BALANCE|              ID|
  +-------------+-------------+----------------+
  |   2021-07-01|     81119.73|         Ax3838J|
  |   2021-07-02|     81119.73|         Ax3838J|
  |   2021-07-03|     81119.73|         Ax3838J|
  |   2021-07-04|     81289.62|         Ax3838J|
  |   2021-07-05|     81385.62|         Ax3838J|
  |   2021-07-02|     81249.76|         Bz3838J|
  |   2021-07-03|     81249.76|         Bz3838J|
  |   2021-07-04|     81249.76|         Bz3838J|
  |   2021-07-05|     81324.28|         Bz3838J|
  |   2021-07-06|     81329.28|         Bz3838J|
  +-------------+-------------+----------------+
  
  </code></pre>
  <p>I can plot for one single <code>ID</code> but I have more than 10000 unique <code>ID</code>s. How can I visualise multiple line plots segmented based on <code>ID</code>. Also, Is there any smart ways to visualise the <code>DF</code> all together?</p>
  <pre><code>DF_single.toPandas().plot.line(x='DATE', y='BALANCE')
  
  </code></pre>
  <p><a href="https://i.stack.imgur.com/vBeiB.png" rel="nofollow noreferrer"><img src="https://i.stack.imgur.com/vBeiB.png" alt="DATE vs BALANCE for single ID" /></a></p>
  <p>Note: Image is for a particular <code>ID</code> from the actual dataset.</p>