Python Excel查询

毕加锁 2022-09-08 我要评论

前言

今天教大家利用Python制作本地Excel的查询与生成的程序

需求

制作一个程序有一个简单的查询入口实现Excel的查询与生成

实验步骤

1打开一个exe 弹出一个界面

2有一个查询卡号点击查询

3下方展示查询的结果同时将这个查询的结果追加到一个新的结果Excel文件里

4新的结果Excel文件格式和源文件格式相同但是每次都在最后追加

今天教大家利用Python制作本地Excel的查询与生成的程序

Excel预览图片

1.2 导入模块并读取Excel文件

等会要用的模块有：pandas、os、xlwt和uuid
用import导入的代码：

import pandas, os, xlwt, uuid

导入好后，就要读取Excel文件了。读取Excel要用到pandas的read_excel函数。

try:
    exl = pandas.read_excel(aim_path)
except:
    print('找不到文件！请检查一下文件路径或文件是否存在')
    os._exit(0)

刚刚导入os模块就是为了做异常捕获找不到文件时的退出。

查询

2.1 Excel的索引与输入

为了方便后面查询，要把DataFrame的索引（index）设为查询输入的卡号。接着，输出以卡号为索引的DF，以便用户查询。最后，就开始循环输入了。

exl.set_index('卡号', inplace = True)
print(f'{exl}\n')
while 1:
    try:
        idx = input('卡号（输入“退出”即可退出）：')
        if idx == '退出':
            os._exit(0)

2.2 开始查询、丰富程序

查询用dataframe.loc[index]来完成，最后输出返回的Series。为了避免用户输入非卡号信息，就又加了异常捕获。

        res = exl.loc[idx]
        print(f'\n{res}\n')
    except KeyError:
        print('你的卡号可能输错了！我找不到这个卡号的人哦~\n')
        continue
    except:
        print('有些错误发生了！\n')
        continue

追加查询结果到Excel

3.1 读取或新建Excel

3.1.1 读取

读取跟上面一样，用read_excel

    try:
        res_exl = pandas.read_excel(res_path)

3.1.2 新建Workbook和Sheet

现在轮到xlwt模块大展身手啦~ 用Workbook函数来新建Workbook；用add_sheet函数新增Sheet

    except:
        workbook = xlwt.Workbook()
        sheet = workbook.add_sheet('new')
        col = 0

3.1.2 写入Column

在Column的位置，需要填入查询的Excel的列索引，用

list(pandas.read_excel(aim_path).columns.values)

可以获取到。然后把列索引以xlwt.write填进去，最后把DF保存再读取这个Excel。

for i in list(pandas.read_excel(aim_path).columns.values):
            sheet.write(0, col, i)
            col += 1
        workbook.save(res_path)
        res_exl = pandas.read_excel(res_path)

3.2 追加结果

首先，把结果res变量设置成列表类型。然后，在这个列表里面新增结果没有的卡号。最后把这个列表设置成一个Series（索引为查询的Excel的列索引）。

    res_series_data = list(res)
    res_series_data.insert(2, idx)
    res_series = pandas.Series(
        res_series_data, 
        index = list(
            pandas.read_excel(aim_path).columns.values
        )
    )

现在建好了Series，准备追加了。追加完后还要保存这个Excel。

    res_exl.loc[str(uuid.uuid1())] = res_series
    try:
        res_exl.to_excel(res_path, index = False)
    except:
        print('写入失败')

这里用了uuid.uuid1来随机产生索引，避免重复而修改其它人的值。最后几行就是保存的操作，python index = False的意思就是把索引隐藏掉了。

完整代码

try:
    exl = pandas.read_excel(aim_path)
except:
    print('找不到文件！请检查一下文件路径或文件是否存在')
    os._exit(0)
exl.set_index('卡号', inplace = True)
print(f'{exl}\n')
while 1:
    try:
        idx = input('卡号（输入“退出”即可退出）：')
        if idx == '退出':
            os._exit(0)
        res = exl.loc[idx]
        print(f'\n{res}\n')
    except KeyError:
        print('你的卡号可能输错了！我找不到这个卡号的人哦~\n')
        continue
    except:
        print('有些错误发生了！\n')
        continue
 
    try:
        res_exl = pandas.read_excel(res_path)
    except:
        workbook = xlwt.Workbook()
        sheet = workbook.add_sheet('new')
        col = 0
        for i in list(pandas.read_excel(aim_path).columns.values):
            sheet.write(0, col, i)
            col += 1
        workbook.save(res_path)
        res_exl = pandas.read_excel(res_path)
    res_series_data = list(res)
    res_series_data.insert(2, idx)
    res_series = pandas.Series(
        res_series_data, 
        index = list(
            pandas.read_excel(aim_path).columns.values
        )
    )
    res_exl.loc[str(uuid.uuid1())] = res_series
    try:
        res_exl.to_excel(res_path, index = False)
    except:
        print('写入失败')