I find VS-code running notebooks with Colab plugin installed pretty helpful for this type of thing already.

(1) run notebook in VS-code https://code.visualstudio.com/docs/datascience/jupyter-noteb...

(2) install Github Copilot extension https://marketplace.visualstudio.com/items?itemName=GitHub.c...

and then it can quite often achieve what you want with just a comment or some previous code snippets.

Then if you load in pandas and a DataFrame you want to analyse, it quite often suggests the next stage of your data-analysis work, either right away, as you write the first step in some chain of pandas methods.