Is your feature request related to a problem? Please describe.
We can improve performance by persisting some additional data.

Describe the solution you would like
.

Component

• Scala Core Internal
• Scala API
• Spark Connect Plugin
• Infrastructure
• PySpark Classic
• PySpark Connect

Additional context

Are you planning on creating a PR?

• I'm willing to make a pull-request