新人求助，筛选出来的表没有标题？

Threebody1

kyimao 发表于 2023-6-2 17:47
报错，这样也试过，将第一行保留，会因为筛选出来的数据找不到对应的列保存，KeyError: " ['序号', '户主 ...

根据错误提示，筛选出来的数据找不到对应的列保存，这是因为原始表格中可能存在其他列，而在筛选后的数据中并不存在这些列名，导致保存时出现了 KeyError 错误。你可以在筛选前，先将原始表格的列名保存下来，然后将该列表作为保留指定列的参数，这样就不会漏掉其他列了。修改后的代码如下所示：

1. import os
   
2. import pandas as pd
   
3. 
   
4. # 原始表格文件夹路径
   
5. input_folder = r'G:\2023\777'
   
6. 
   
7. # 保存新表格的文件夹路径
   
8. output_folder = r'G:\2023\555'
   
9. 
   
10. # 获取原始表格文件列表
    
11. input_files = os.listdir(input_folder)
    
12. 
    
13. for file in input_files:
    
14.     if file.endswith('.xls'):
    
15.         # 构建原始表格文件的完整路径
    
16.         file_path = os.path.join(input_folder, file)
    
17.         
    
18.         # 读取原始表格数据，保留第一行作为标题行
    
19.         df = pd.read_excel(file_path, header=0)
    
20.         
    
21.         # 保存原始表格的列名
    
22.         columns_to_keep = df.columns.tolist()
    
23.         
    
24.         # 筛选出xx社区
    
25.         filtered_df = df[df.iloc[:, 6] == 'xx社区']
    
26.         
    
27.         # 重新排列序号列
    
28.         filtered_df.loc[:, '序号'] = range(1, len(filtered_df) + 1)
    
29.         
    
30.         # 保留原始表格的列名
    
31.         filtered_df = filtered_df[columns_to_keep]
    
32.         
    
33.         # 保留指定列
    
34.         columns_to_keep = ['序号', '户主姓名', 'xx社区','金额合计']
    
35.         filtered_df = filtered_df[columns_to_keep]
    
36.         
    
37.         # 创建新表格的文件名
    
38.         output_file = os.path.join(output_folder, os.path.splitext(file)[0] + '.xlsx')
    
39.         
    
40.         # 保存新表格，加上标题行
    
41.         with pd.ExcelWriter(output_file) as writer:
    
42.             df.to_excel(writer, sheet_name='Sheet1', index=False)
    
43.             filtered_df.to_excel(writer, sheet_name='Sheet1', index=False, startrow=1)

复制代码

这样就可以同时保留原表标题，并成功生成新表格了。
（我是他小号，但是他每小时限额到了，我帮忙）

Threebody1

sfqxx 发表于 2023-6-2 17:53
可以在读取原表数据时，将第一行作为表头(header)保存下来，然后在生成新表格时添加回去。

修改代码如下 ...

Threebody1

sfqxx 发表于 2023-6-2 18:39