(1)需求:将数据按A列值分成多个文件夹,再按B列值分成多个Excel表格,并将数据存到对应文件夹内。
(2)实现
import pandas as pd
import os
# 读取数据
df = pd.read_excel('C:/Users/Desktop/data.xlsx')
# 按A列进行分组
groups = df.groupby('A列')
# 遍历每个分组
for group_name, group_data in groups:
# 创建文件夹
folder_name = str(group_name)
folder_path = os.path.join('D:\\', folder_name)
#使用os.mkdir()方法在D盘上创建一个文件夹,该文件夹名为分组名
os.mkdir(folder_path)
# 按B列的相同值将数据分组,并保存到Excel文件中
subgroups = group_data.groupby('B列')
for subgroup_name, subgroup_data in subgroups:
# 生成Excel文件名
file_name = f"{subgroup_name}.xlsx"
#os.path.join:路径拼接作用
file_path = os.path.join(folder_path, file_name)
# 将数据保存到Excel文件中
subgroup_data.to_excel(file_path, index=False)