autoplotlib
v0.1.1
通过简单地通过文本描述它们,在Python中迅速生成图。例如,只需使用提示“创建带有名称标签的散点图”,而不是无休止地搜索文档和stackoverflow。鉴于这样的提示, autoplotlib Python库使用OpenAI API自动生成相应的代码并绘图 - 节省了时间和精力。
警告此软件包是实验性的。该软件包可以从语言模型中执行代码输出。尽管不太可能,但始终检查并确认在执行前没有生成该模型的恶意代码。自行使用。
pip install autoplotlib
以下是使用autoplotlib库绘制PANDAS数据框架的示例。 plot()函数采取了一个提示,描述了绘图和数据绘图。该函数返回绘图,生成的代码以及OpenAI API的完整响应。
# set the OpenAI API key
import os
os . environ [ "OPENAI_API_KEY" ] = "<YOUR_API_KEY>"
import autoplotlib as aplt
import pandas as pd
data = pd . DataFrame (
[[ 29 , 177 ], [ 33 , 186 ], [ 48 , 161 ], [ 53 , 173 ], [ 67 , 152 ]],
index = [ "Alice" , "Bob" , "Charlie" , "Dave" , "Eve" ],
columns = [ "age" , "height" ],
)
figure_description = """
Plot the data as scatterplot between height and age.
Add the names as labels next to the data points.
Ensure the labels don't overlap.
Mark people taller than 170 with a star instead of a point.
"""
code , fig , llm_response = aplt . plot ( figure_description , data = data )有关贡献的信息,请参见开发指南。
麻省理工学院许可证