Pandas AI 是一个 Python 库，它为流行的数据分析和操作工具 Pandas 添加了生成人工智能功能。
用法
PandasAI 旨在与 pandas 结合使用。它使 Pandas 具有对话性，允许以 pandas DataFrames 的形式提出有关数据的问题并获得答案。例如可以要求 PandasAI 查找 DataFrame 中列值大于 5 的所有行，它将返回仅包含这些行的 DataFrame：
import pandas as pd
from pandasai import PandasAI
# Sample DataFrame
df = pd.DataFrame({
"country": ["United States", "United Kingdom", "France", "Germany", "Italy", "Spain", "Canada", "Australia", "Japan", "China"],
"gdp": [19294482071552, 2891615567872, 2411255037952, 3435817336832, 1745433788416, 1181205135360, 1607402389504, 1490967855104, 4380756541440, 14631844184064],
"happiness_index": [6.94, 7.16, 6.66, 7.07, 6.38, 6.4, 7.23, 7.22, 5.87, 5.12]
})
# Instantiate a LLM
from pandasai.llm.openai import OpenAI
llm = OpenAI(api_token="YOUR_API_TOKEN")
pandas_ai = PandasAI(llm, conversational=False)
pandas_ai(df, prompt='Which are the 5 happiest countries?')
上面的代码将返回以下内容：
6 Canada
7 Australia
1 United Kingdom
3 Germany
0 United States
Name: country, dtype: object
当然也可以让 PandasAI 进行更复杂的查询。例如可以要求 PandasAI 求出 2 个最不幸福国家的 GDP 总和：
pandas_ai(df, prompt='What is the sum of the GDPs of the 2 unhappiest countries?')
上面的代码将返回以下内容：
19012600725504
也可以请 PandasAI 画图：
pandas_ai(
df,
"Plot the histogram of countries showing for each the gpd, using different colors for each bar",
)
命令行工具
Pai 是一个命令行工具，旨在提供一种通过命令行界面 (CLI) 与 PandasAI 交互的便捷方式。
pai [OPTIONS]
选项：
-
-d, –dataset：数据集的文件路径。 -
-t, –token：HuggingFace 或 OpenAI API 令牌，如果没有提供令牌，pai 将从.env 文件中提取。 -
-m, –model : LLM 的选择，openai, open-assistant, 或 starcoder. -
-p, –prompt : 提示 PandasAI 将运行。
要查看可用选项及其描述的完整列表，请运行以下命令：
pai --help
例如
pai -d "~/pandasai/example/data/Loan payments data.csv" -m "openai" -p "How many loans are from men and have been paid off?"
应产生与 from_csv.py 示例相同的输出。
环境变量
为了设置 LLM（Hugging Face Hub，OpenAI）的 API 密钥，需要设置适当的环境变量。可以通过将.env.example 文件复制到.env：
cp .env.example .env
编辑.env 文件并设置适当的值。或者可以将环境变量直接传递给 LLM 的构造函数：
# OpenAI
llm = OpenAI(api_token="YOUR_API_KEY")
# Starcoder
llm = Starcoder(api_token="YOUR_HF_API_KEY")
传送门
开源协议：Apache2.0
开源地址：https://github.com/gventuri/pandas-ai
项目合集：https://github.com/OpenTechCol/OpenTechCol
「回复【加群】加入开源技术交流群，干货很多！」
-END-
原文始发于微信公众号（开源技术专栏）：Pandas AI 支持人工智能的Python库！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 举报，一经查实，本站将立刻删除。
文章由极客之音整理，本文链接：https://www.bmabk.com/index.php/post/145736.html
