Python实现Excel数据过滤

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