【Python模块】pandas-AI火了！

大家好，我是Peter~

关注小屋的朋友肯定都知道小编一直在坚持写pandas相关的文章，最近的一篇请参考：

Pandas是一个基于python和numpy的数据分析库，特别擅长数据处理和数据分析。

今年AI在大模型方面十分火热，尤其是ChatGPT等工具的出现，让我们的编程更加智（失）能（业）。

最近，就有一位大佬将Pandas和AI强强联手，帮助我们更加快速便捷地分析数据。

GitHub项目地址：github.com/gventuri/pa…

Pandas-ai现身

Github官方解释什么是pandas-ai：

Pandas AI is a Python library that adds generative artificial intelligence capabilities to Pandas, the popular data analysis and manipulation tool. It is designed to be used in conjunction with Pandas, and is not a replacement for it.

Pandas AI 是一个 Python 库，它为流行的数据分析和操作工具 Pandas 添加了生成人工智能功能。它旨在与 Pandas 结合使用，而不是它的替代品。

安装十分简单：

pip install pandasai  # 命令行
!pip install pandasai  # jupyter notebook中

使用

1、导入相关的库：

import pandas as pd
from pandasai import PandasAI
from pandasai.llm.openai import OpenAI

2、模拟生成数据：

import pandas as pd
from pandasai import PandasAI

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]
})

3、初始化一个llm模型

from pandasai.llm.openai import OpenAI
llm = OpenAI()

4、调用大模型实例，使用提示promote

pandas_ai = PandasAI(llm)
pandas_ai.run(df, prompt='Which are the 5 happiest countries?')

显示结果为：

6            Canada
7         Australia
1    United Kingdom
3           Germany
0     United States
Name: country, dtype: object

另一个实例：

pandas_ai.run(df, prompt='What is the sum of the GDPs of the 2 unhappiest countries?')

19012600725504

绘图功能也是轻松搞定：

pandas_ai.run(
    df,
    "Plot the histogram of countries showing for each the gpd, using different colors for each bar",
)

---------------------------END---------------------------

题外话

感兴趣的小伙伴，赠送全套Python学习资料，包含面试题、简历资料等具体看下方。

👉CSDN大礼包🎁：全网最全《Python学习资料》免费赠送🆓！（安全链接，放心点击）

一、Python所有方向的学习路线

Python所有方向的技术点做的整理，形成各个领域的知识点汇总，它的用处就在于，你可以按照下面的知识点去找对应的学习资源，保证自己学得较为全面。

二、Python必备开发工具

工具都帮大家整理好了，安装就可直接上手！

三、最新Python学习笔记

当我学到一定基础，有自己的理解能力的时候，会去阅读一些前辈整理的书籍或者手写的笔记资料，这些笔记详细记载了他们对一些技术点的理解，这些理解是比较独到，可以学到不一样的思路。

四、Python视频合集

观看全面零基础学习视频，看视频学习是最快捷也是最有效果的方式，跟着视频中老师的思路，从基础到深入，还是很容易入门的。

五、实战案例

纸上得来终觉浅，要学会跟着视频一起敲，要动手实操，才能将自己的所学运用到实际当中去，这时候可以搞点实战案例来学习。

六、面试宝典

简历模板

👉CSDN大礼包🎁：全网最全《Python学习资料》免费赠送🆓！（安全链接，放心点击）

若有侵权，请联系删除