Using Gemini code assistant you can open your data file or give a sample of your data, describe what you want to do with the data, Gemini can generate the code for you. The generated code doesn't have to work though, but if you test and give Gemini feedback, it will help you debug and improve the code. I guess the data science agent can make the task even easier.