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 )有關貢獻的信息,請參見開髮指南。
麻省理工學院許可證