Python实现Excel数据过滤
作者：matrixlzp
本文将以车牌数据为示例,为大家详细介绍一下如何使用Python实现Excel数据过滤功能,感兴趣的小伙伴可以跟随小编一起学习一下
一、场景分析
假设有如下一份 excel 数据 shop.xlsx, 写一段 python 程序，实现对于车牌的分组数据过滤。
并以车牌为文件名，把店名输出到 车牌.txt 文件中。
比如 闽A.txt 文件内容为：
小林书店福州店1
小林书店福州店2
二、依赖安装
程序依赖 pandas 对 excel 数据进行处理，所以需要先安装一下 pandas 依赖。
pip install pandas
三、代码实现
import pandas as pd
# pandas 菜鸟教程
# https://www.runoob.com/pandas/pandas-dataframe.html
# 输入文档
input_path = r"C:\Users\Administrator\Desktop\py\excel\filter\shop.xlsx"
# 读取 Excel 文件，将其存储在一个DataFrame对象中
df = pd.read_excel(input_path)
# 行数，注意这个行数是数据行数，标题不算
rows = df.shape[0]
# 最终的数据结果是，{ '闽A':[], '闽B':[], '闽C':[] }
result = {}
# 根据 车牌号，获取店名
for i in range(0, rows):
# df.loc[ idx, 'Column1']
# idx : 行标，从 0 开始，0 是数据行的第一行，即 excel 的第二行
carNo = df.loc[i, '车牌']
print( carNo )
shopName = df.loc[i, '店名']
print( shopName )
# 以 carNo 为 key 从字典中获取值列表，没有返回 None，避免 KeyError
value_list = result.get(carNo, None)
if value_list:
value_list.append(shopName)
else:
value_list = []
value_list.append(shopName)
result[ carNo ] = value_list
print( result )
# 根据 carNo, 生成 txt, 内容是 店名
# 遍历 key 列表
for key in result.keys():
value_list = result.get(key)
output_path = key+'.txt'
# 以 write 的方式打开输出文件
with open( output_path, 'w', encoding='utf-8') as file:
num = len( value_list )
end = num - 1
for i, item in enumerate(value_list):
# 最后一行数据不需要换行
if i == end:
file.write(item)
else:
file.write(item + '\n')四、运行
py shop_filter.py
到此这篇关于Python实现Excel数据过滤的文章就介绍到这了,更多相关Python Excel数据过滤内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！