pandas-ai

环境要求

• Python 版本 3.8+ 且不超过 3.11
• 已安装 pip 或 poetry 包管理器
• 一个可用的 LLM API Key（示例使用 OpenAI，如 gpt-4.1-mini）
• 可选的 Docker 环境（仅使用 Docker 沙箱时需要）

安装与启动步骤

1. 1安装 PandasAI

   用 pip 安装核心库和 LiteLLM 适配包，须在 Python 3.8~3.11 环境中执行。

   pip install pandasai
   pip install pandasai-litellm

   复制

2. 2或用 Poetry 安装

   若项目用 Poetry 管理依赖，改用 poetry add 添加同样两个包即可。

   poetry add pandasai
   poetry add pandasai-litellm

   复制

3. 3初始化 LLM

   引入 pandasai 与 LiteLLM，用你的 OpenAI 密钥创建模型实例，再注入到配置中。

   import pandasai as pai
   from pandasai_litellm.litellm import LiteLLM
   
   # Initialize LiteLLM with your OpenAI model
   llm = LiteLLM(model="gpt-4.1-mini", api_key="YOUR_OPENAI_API_KEY")
   
   # Configure PandasAI to use this LLM
   pai.config.set({
       "llm": llm
   })

   复制

4. 4读数据并提问

   用 pai.read_csv 读取 CSV，再调用 df.chat 用自然语言提问并打印回答。

   # Load your data
   df = pai.read_csv("data/companies.csv")
   
   response = df.chat("What is the average revenue by region?")
   print(response)

   复制

5. 5可选 Docker 沙箱

   需要隔离环境安全执行生成代码时，安装 pandasai-docker 并导入 DockerSandbox。

   pip install "pandasai-docker"

   复制

关键配置

如何确认成功

运行脚本后终端打印出 df.chat 的自然语言回答（例如平均营收数值），即表示库与 LLM 配置成功。

常见问题

注意事项

• 代码示例中的 api_key 为占位符，请替换为真实密钥，切勿提交到公开仓库。
• Python 版本必须满足 3.8+ <=3.11。
• pandasai-litellm 是 LLM 适配包，需与 pandasai 一起安装才能按示例运行。
• Docker 沙箱为可选项，仅在需要隔离执行生成代码时安装 pandasai-docker。