Skip to content

PlotAI - Your Ultimate Plotting Assistant! πŸ“ŠπŸ€– Use ChatGPT-3.5 to create plots in Python and Matplotlib directly in your Python script or notebook.



Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?

Latest commit


Git stats


Failed to load latest commit information.
Latest commit message
Commit time
September 18, 2023 11:24
August 24, 2023 18:11
August 24, 2023 18:11
August 23, 2023 17:17
August 23, 2023 14:45
August 23, 2023 14:45
August 25, 2023 13:55
October 3, 2023 16:27
August 25, 2023 15:23

PlotAI logo

πŸŽ¨πŸ€– Create Python plots in matplotlib with LLM πŸŽ¨πŸ€–

πŸš€ Get Started Β Β β€’Β Β  🀝 Issues Β Β β€’Β Β  🐦 Twitter Β Β β€’Β Β  πŸ‘©β€πŸ’Ό LinkedIn

PlotAI πŸŽ¨πŸ€–

The easiest way to create plots in Python and Matplotlib. The plotai is using LLM to generate code and plots.

The idea:

  1. User provides input DataFrame and prompt.
  2. The PlotAI constructs a prompt for LLM, which contains the first 5 rows of DataFrame and the user's prompt and asks for Python code as output.
  3. Returned Python code is executed, and the plot is displayed.

The simplest possible API for plotting:

# import packages
import pandas as pd
from plotai import PlotAI
# create some data
df = pd.DataFrame({"x":[1,2,3], "y": [4,5,6]})
# do a plot
plot = PlotAI(df)
plot.make("scatter plot")

The PlotAI class has only one method, make().

It works in Python scripts and in notebooks (Jupyter, Colab, VS Code).

PlotAI logo

πŸš€ Get started

Install plotai package:

pip install plotai

Create .env file with the OpenAI API key:


You can also pass the OpenAI API key in Python:

import os
os.environ["OPENAI_API_KEY"] = "your-key-here"

Import plotai and make plots:

# import PlotAI
from plotai import PlotAI

# create PlotAI object, pass pandas DataFrame as an argument
plot = PlotAI(df)

# make a plot, just tell what you want
plot.make("make a scatter plot")

More examples

Analyze the GPD dataset


Analyze the Iris dataset

πŸ‘©β€πŸ’ΌπŸ¦ Connect with Us on LinkedIn & Twitter

Stay up-to-date with the latest updates about PlotAI πŸŽ¨πŸ€– by following us on Twitter (MLJAR Twitter) and LinkedIn (Aleksandra LinkedIn & Piotr LinkedIn). We look forward to connecting with you and hearing your thoughts, ideas, and experiences with PlotAI. Let's explore the future of AI together!

⚠️ Limitations

PlotAI is in very experimental form, below are some limitations:

  • PlotAI is using OpenAI ChatGPT-3.5-turbo for completions, it will be nice to extend to other LLMs.
  • PlotAI is sending 5 first rows from your DataFrame to OpenAI ChatGPT. If you have sensitive data, please remove/encode it before passing to PlotAI.
  • PlotAI is executing Python code returned by LLM, it can be dangerous and unsafe. It would be nice to have the option to accept the response code before execution.

πŸ›‘ Disclaimer

This project, PlotAI, is provided "as is" without any warranty, express or implied. By using this software, you agree to assume all risks associated with its use, including but not limited to data loss, system failure, or any other issues that may arise. The developers and contributors of this project do not accept any responsibility or liability for any losses, damages, or other consequences that may occur as a result of using this software.

Please note that the use of the OpenAI language models can be expensive due to its token usage. By utilizing this project, you acknowledge that you are responsible for monitoring and managing your own token usage and the associated costs. It is highly recommended to check your OpenAI API usage regularly and set up any necessary limits or alerts to prevent unexpected charges.


PlotAI - Your Ultimate Plotting Assistant! πŸ“ŠπŸ€– Use ChatGPT-3.5 to create plots in Python and Matplotlib directly in your Python script or notebook.








No releases published