SuperAPI AI 工具集

pandasai

数据与分析

查看项目

更新于 10-16 03:39

技能介绍

用途:PandasAI 让你用自然语言与数据对话,对 pandas DataFrame 提问并获得洞察、可视化与解释,由 LLM 驱动。

主要能力:通过 smart_df.chat(question) 进行自然语言查询,返回数值、DataFrame 或图表;用 SmartDatalake([df1, df2]).chat(q) 做多表查询(需先设置 df.name);用 smart_df.last_code_generated 查看生成的代码以验证正确性;支持缓存(enable_cache)、隐私模式(enforce_privacy)与图表自动保存(save_charts、save_charts_path)。支持 OpenAI、Anthropic、Google Gemini、Azure OpenAI 及本地 Ollama 等后端,通过 config 传入不同 llm 对象切换。

使用方式:安装命令为 uv add pandasai,按需追加 openai 或 matplotlib seaborn plotly;设置对应环境变量如 OPENAI_API_KEY。基本用法是读取 CSV 后创建 SmartDataframe 并调用 chat。隐私场景可先对敏感列做哈希匿名化再启用 enforce_privacy。

适用场景:探索陌生数据集、由描述性提示快速生成可视化、向相关方解释复杂数据转换、构建对话式数据探索界面(Streamlit、Jupyter、FastAPI)、快速原型化分析流程。

限制:不适合需要确定性、版本可控输出的生产管道;处理高度敏感 PII 需先匿名化;超过 10 万行的 DataFrame 性能受限,建议先采样到 1 万行以内;简单查询直接用 pandas 更快。注意设置 temperature=0 以获得可复现结果,缓存按问题字符串为键、数据变化不会自动失效,无头环境需在导入前将 matplotlib 后端设为 Agg。完整示例与集成模式见其参考资料与上游文档。