32.单一查询(文心一言&智普AI

李光朱课程 2024-03-30 01:34:24

目前市面上AI的工具越来越多,今天我们把同样的需求放到不同的AI工具上执行。对比一下结果。

案例导入

如下图所示,工作表中A1:D5为给定数据,A-D列分别为“编号”、“姓名”、“额度”和“名次”数据。用pandas导入Excel文件的数据,第1行为索引行,只导入前5行数据,引擎为"openpyxl",查询编号3,1和2对应的全部行数据并输出。

图1.原始数据表

提示词

文心一言和智普AI我们使用同样的提示词,对比双方给出的结果。提示词如下:“你是pandas专家,文件路径为:D:/Samples/个人信息-查询.xlsx。该Excel文件的第一个工作表中A1:D5为给定数据,A-D列分别为“编号”、“姓名”、“额度”和“名次”数据。用pandas导入Excel文件的数据,第1行为索引行,只导入前5行数据,引擎为"openpyxl",查询编号3,1和2对应的全部行数据并输出。”

文心一言结果

首先,你需要确保已经安装了pandas和openpyxl库。如果没有安装,可以通过pip进行安装:

pip install pandas openpyxl

然后,你可以使用以下Python代码来导入Excel文件的数据,设置第一行为索引,只导入前5行数据,并使用openpyxl作为引擎。最后,你将查询编号3、1和2对应的全部行数据并输出。

import pandas as pd# 设置文件路径file_path = 'D:/Samples/个人信息-查询.xlsx'# 使用pandas读取Excel文件# 设置header=0表示第一行为列名,nrows=5表示只读取前5行数据,engine='openpyxl'表示使用openpyxl引擎df = pd.read_excel(file_path, header=0, nrows=5, engine='openpyxl')# 查询编号3, 1和2对应的行数据# 使用isin函数查询多个值filtered_df = df[df['编号'].isin([3, 1, 2])]# 输出查询结果print(filtered_df)

确保你的Excel文件路径、工作表名称和列名与上述代码中的设置相匹配。如果列名或文件路径有所不同,你需要相应地调整代码中的参数。

运行上述代码后,你会看到包含编号3、1和2的行数据的输出。如果这些编号在你的数据集中不存在,那么filtered_df将是一个空DataFrame。

智普AI结果

图2.智普AI结果1

图3.智普AI结果2

运行结果

图4.运行结果对比图

从结果看,智普AI的结果是按照3、1、2的顺序排列,而文心一言的是按1、2、3的顺序排列。

本文使用的工具如下:

文心一言:3.5版本

智普AI Ask CodeGeeX插件

Python编辑器:3.9.0

申明:以上结果皆为实际运行结果,如遇到无法运行可在评论区留言。部分数据来自网络,如有侵权请私信告知,感谢!

0 阅读:0

李光朱课程

简介:感谢大家的关注