plotai

PlotAI 项目介绍 🎨🤖

项目背景

PlotAI 是一个用于在 Python 和 Matplotlib 中创建可视化图表的工具，其特别之处在于它使用了大型语言模型（LLM）来自动生成代码和图表。这意味着用户只需提供一个数据框和简单的提示，即可得到相应的图表展示。这种简单直观的方式，大大降低了 Python 绘图的门槛。

功能理念

PlotAI 的基本工作流程如下：

1. 用户提供输入数据框（DataFrame）和绘图提示。
2. PlotAI 生成一个包含数据框前五行数据和用户提示的请求，并请求 LLM 返回相应的 Python 代码。
3. 获得的 Python 代码被执行，生成并展示绘图结果。

通过以上三个步骤，用户可以轻松创建各种类型的图表，只需一个简单的提示即可。

简单的示例

PlotAI 提供了简洁的 API，以下是一个快速创建散点图的示例：

# 导入必要的包
import pandas as pd
from plotai import PlotAI

# 创建一些数据
df = pd.DataFrame({"x":[1,2,3], "y":[4,5,6]})

# 使用 PlotAI 创建图表
plot = PlotAI(df)
plot.make("scatter plot")

PlotAI 类仅有一个方法 make()，可在 Python 脚本与 Jupyter、Colab、VS Code 等笔记本环境下使用。

如何开始

首先，需要安装 PlotAI 包：

pip install plotai

然后，请确保你拥有 OpenAI 的 API 密钥，并在 .env 文件中配置或在 Python 中设置：

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

之后，便可以导入 PlotAI 并进行绘图：

# 导入 PlotAI
from plotai import PlotAI

# 创建 PlotAI 对象，并传入数据框作为参数
plot = PlotAI(df)

# 进行绘图，只需说明你想要的图表类型
plot.make("make a scatter plot")

PlotAI 默认使用 gpt-3.5-turbo 模型，也可以选择使用其他 OpenAI 模型，例如 gpt-4。

更多示例

• 分析 GDP 数据集
• 分析鸢尾花数据集

社交媒体联系

可以通过 Twitter 和 LinkedIn 关注 PlotAI 的最新动态，加入我们的讨论，与我们一同探索 AI 的未来：

• MLJAR Twitter
• Aleksandra LinkedIn
• Piotr LinkedIn

限制

PlotAI 当前仍处于实验阶段，存在以下限制：

• 目前仅支持使用 OpenAI 的 ChatGPT-3.5-turbo，希望未来拓展至更多 LLM。
• 将数据框的前五行数据发送给 OpenAI ChatGPT，请在传递数据前确保敏感信息已删除或编码。
• LLM 返回的 Python 代码会被直接执行，这存在一定的安全风险。未来可能会提供代码审核选项。

免责声明

PlotAI 项目提供 "按原样" 使用，未附带任何明示或暗示的担保。用户在使用软件时，自行承担所有风险。开发者不对其使用过程中可能引发的任何数据损失、系统故障或其他问题负责。

使用 OpenAI 模型可能产生费用，建议用户定期监控 API 使用量并设置必要的限制和提醒。

其他项目

我们正在开发 Mercury 框架，用于将 Python 笔记本共享为交互式 Web 应用。提供 一键部署 Python 笔记本 的服务，欢迎访问 网站 了解更多信息。