BimAnt

PlotAI = Matplotlib + AI

admin -

PlotAI是在 Python 和 Matplotlib 中创建绘图的最简单方法，它使用 LLM 生成Python代码和Matplotlib绘图。

1、PlotAI简介

PlotAI的思路：

  • 用户提供输入数据框和提示。
  • PlotAI 构建了 LLM 的提示，其中包含 DataFrame 的前 5 行和用户的提示，并要求 Python 代码作为输出。
  • 执行返回的 Python 代码，并显示绘图。

下面是一个简单的示例：

# 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")

PlotAI 类只有一个方法： make() ，它适用于 Python 脚本和笔记本（Jupyter、Colab、VS Code）。

2、PlotAI快速上手

安装plotai包：

pip install plotai

使用OpenAI的API key创建 .env 文件：

OPENAI_API_KEY=your-key-here

也可以在Python代码中设置OpenAI的API key：

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

导入plotai，然后制作图表：

# 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")

3、PlotAI示例

下面的示例使用PlotAI绘制iris数据集的图表：

4、已知的限制

PlotAI 仍处于实验阶段，以下是一些限制：

  • PlotAI 使用 OpenAI ChatGPT-3.5-turbo 来完成，如果可以扩展到其他LLM，那就太好了。
  • PlotAI 将 DataFrame 中的前 5 行发送到 OpenAI ChatGPT。 如果你有敏感数据，请在传递到 PlotAI 之前将其删除/编码。
  • PlotAI 执行 LLM 返回的 Python 代码，这可能是危险且不安全的。 如果能够选择在执行之前接受响应代码，那就太好了。

原文链接：Create plots in Python and Matplotlib with LLM

BimAnt翻译整理，转载请标明出处