Meet Sketch: An AI code Writing Assistant For Pandas

For pandas users, Sketch is an AI code-writing helper that substantially comprehends your data’s context to increase ideas’ relevance. The Sketch is immediately useable and doesn’t need to be added as a plugin to your IDE. Using machine learning and natural language processing, the AI code-writing helper Sketch helps programmers write code. It can generate entire functions, complete statements, and code snippets based on a description of what the code should accomplish. With Sketch, developers will be able to create code more quickly and with less effort, allowing them to concentrate on more difficult problems.

Sketch can also automate repetitive tasks, find bugs and suggest fixes, analyze the codebase, and offer suggestions for optimization. Additionally, it can facilitate code reworking and enhance code maintainability. Since Sketch integrates with many code editors and IDEs, it is available to various developers. Sketch integrates with several code editors, including Atom, Visual Studio Code, and Sublime Text.

How does it function?

Modern AI methods like deep learning and reinforcement learning are used by Sketch to enhance its code writing recommendations over time. Additionally, it has an intuitive user interface that makes it simple for developers of all experience levels to use it immediately. Further, Sketch is always learning new things and expanding its knowledge base to stay current with the latest programming languages and technologies. Programming languages like JavaScript, Python, Java, C++, and more can all be used with the AI code-writing assistant.

The only required steps are importing Sketch and adding the .sketch extension to any Pandas data frame.


Ask is a straightforward question-and-answer feature on Sketch; it will provide a textual response based on the data’s summary statistics and description.

Ask can be used to learn more about the data, develop better column names, pose hypothetical questions like, “How would I go about performing X with this data?” and more.


The fundamental “code-writing” prompt in Sketch is howto. This will provide a code block that you may copy and paste to serve as the basis (or even the conclusion!) of any queries you have for the data. Ask them how to normalize, develop new features, plot, and even build models after they’ve cleaned up the data.


A more sophisticated prompt that is better for data generation is .apply

.sketch.apply use it to generate new features, parse fields, and more. The foundation for this is lambda prompt. To utilize this, you will need to create a free OpenAI account and establish an environment variable with your API key. OPENAI API KEY=YOUR API KEY


  • Using Sketch as an AI writing assistant for code results in increased productivity, fewer errors, and better code quality.
  • Sketch can help inexperienced developers learn new programming languages and best practices, making it a valuable tool for both experienced and novice developers. • By automating routine tasks and offering suggestions, Sketch can save developers valuable time and allow them to focus on more complex and challenging problems.
  • Sketch can handle numerous programming paradigms, including object-oriented, functional, and procedural programming, and it interfaces with well-known collaboration and version control technologies like Git and GitHub. It is intended to adjust to the individual coding style of the developer and can eventually learn from their preferences.
  • The AI code writing assistance can also spot trends in the code and suggest how to modify it to make it easier to read and maintain.
  • Sketch offers in-the-moment feedback, which can assist developers in identifying flaws and averting costly errors.


  • It depends on the calibre of the data it was trained on and might not always make the best recommendations.
  • Depending on how hard the task at hand is, its recommendations’ correctness may change.
  • It could take some time for developers to become proficient with Sketch and adapt to its recommendations.
  • Developers must still apply their knowledge and expertise in addition to Sketch to make wise selections.

Despite these drawbacks, Sketch can dramatically increase developers’ productivity, individually and collectively. Developers can produce better, more effective code and keep up with the newest programming languages and technologies by utilizing AI technology. Sketch will probably continue to improve and provide developers with even more benefits as AI technology develops.

In conclusion, Sketch is a useful tool for developers who want to increase their coding speed, accuracy, and quality. By leveraging AI technology, it offers practical advice and automates repetitive activities, freeing up engineers to concentrate on more challenging issues.

Check out the Github. All Credit For This Research Goes To the Researchers on This Project. Also, don’t forget to join our 13k+ ML SubRedditDiscord Channel, and Email Newsletter, where we share the latest AI research news, cool AI projects, and more.

Dhanshree Shenwai is a Computer Science Engineer and has a good experience in FinTech companies covering Financial, Cards & Payments and Banking domain with keen interest in applications of AI. She is enthusiastic about exploring new technologies and advancements in today’s evolving world making everyone's life easy.

🐝 Join the Fastest Growing AI Research Newsletter Read by Researchers from Google + NVIDIA + Meta + Stanford + MIT + Microsoft and many others...