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

kyimao

import os
import pandas as pd

# 原始表格文件夹路径
input_folder = r'G:\2023\777'

# 保存新表格的文件夹路径
output_folder = r'G:\2023\555'

# 获取原始表格文件列表
input_files = os.listdir(input_folder)

for file in input_files:
    if file.endswith('.xls'):
        # 构建原始表格文件的完整路径
        file_path = os.path.join(input_folder, file)
        
        # 读取原始表格数据，跳过第一行
        df = pd.read_excel(file_path, skiprows=1)
        
        # 筛选出xx社区
        filtered_df = df[df.iloc[:, 6] == 'xx社区']
        
        # 重新排列序号列
        filtered_df.loc[:, '序号'] = range(1, len(filtered_df) + 1)
        
        # 保留指定列
        columns_to_keep = ['序号', '户主姓名', 'xx社区','金额合计']
        filtered_df = filtered_df[columns_to_keep]
        
        # 创建新表格的文件名
        output_file = os.path.join(output_folder, os.path.splitext(file)[0] + '.xlsx')
        
        # 保存新表格，跳过第一行
        filtered_df.to_excel(output_file, index=False, startrow=1)
上面代码是将文件夹里一堆表，按社区名字进行筛选后，生成新的表，但是生成出来的，没有标题，但是我又想保留原表标题，我该怎么搞？

歌者文明清理员

你可以在读取原始表格数据时，保留第一行作为标题行，而不是跳过第一行，例如使用header=0参数表示将第一行作为标题行，然后在保存新表格时，在原来的基础上加上标题行即可。修改后的代码如下所示：

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.         # 筛选出xx社区
    
22.         filtered_df = df[df.iloc[:, 6] == 'xx社区']
    
23.         
    
24.         # 重新排列序号列
    
25.         filtered_df.loc[:, '序号'] = range(1, len(filtered_df) + 1)
    
26.         
    
27.         # 保留指定列
    
28.         columns_to_keep = ['序号', '户主姓名', 'xx社区','金额合计']
    
29.         filtered_df = filtered_df[columns_to_keep]
    
30.         
    
31.         # 创建新表格的文件名
    
32.         output_file = os.path.join(output_folder, os.path.splitext(file)[0] + '.xlsx')
    
33.         
    
34.         # 保存新表格，加上标题行
    
35.         with pd.ExcelWriter(output_file) as writer:
    
36.             df.to_excel(writer, sheet_name='Sheet1', index=False)
    
37.             filtered_df.to_excel(writer, sheet_name='Sheet1', index=False, startrow=1)

复制代码

这样就可以保留原表标题，并生成新表格了。

kyimao

歌者文明清理员 发表于 2023-6-2 17:13
你可以在读取原始表格数据时，保留第一行作为标题行，而不是跳过第一行，例如使用header=0参数表示将第一行 ...

报错，这样也试过，将第一行保留，会因为筛选出来的数据找不到对应的列保存，KeyError: " ['序号', '户主姓名', 'xx社区','金额合计'] not in index"

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)

复制代码

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

sfqxx

可以在读取原表数据时，将第一行作为表头(header)保存下来，然后在生成新表格时添加回去。

修改代码如下：

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

复制代码

Threebody1

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

修改代码如下 ...

isdkz

在这种情况下，您希望在筛选数据后，保留原表的标题。这可以通过在读取原始表格时，不跳过标题行来实现。然后在保存新表格时，不需要再跳过第一行。具体修改如下：

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)
    
20.         
    
21.         # 筛选出xx社区
    
22.         filtered_df = df[df.iloc[:, 6] == 'xx社区']
    
23.         
    
24.         # 重新排列序号列
    
25.         filtered_df.loc[:, '序号'] = range(1, len(filtered_df) + 1)
    
26.         
    
27.         # 保留指定列
    
28.         columns_to_keep = ['序号', '户主姓名', 'xx社区','金额合计']
    
29.         filtered_df = filtered_df[columns_to_keep]
    
30.         
    
31.         # 创建新表格的文件名
    
32.         output_file = os.path.join(output_folder, os.path.splitext(file)[0] + '.xlsx')
    
33.         
    
34.         # 保存新表格，不跳过第一行
    
35.         filtered_df.to_excel(output_file, index=False)
    

复制代码

这样修改后，新的表格就会有原始表格的标题了。注意，这种方法假定原始表格的标题位于第一行，并且列名为'序号', '户主姓名', 'xx社区', '金额合计'。如有不同，请进行相应调整。

sfqxx

Threebody1 发表于 2023-6-2 17:56

Threebody1

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

kyimao

isdkz 发表于 2023-6-2 18:06
在这种情况下，您希望在筛选数据后，保留原表的标题。这可以通过在读取原始表格时，不跳过标题行来实现。然 ...

不行，原始表中有多个列，但是我只需要，这几个列，按你的代码，输出时就会提示找不到对应列名

kyimao

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

修改代码如下 ...

一样提示找不到xx社区，这列

sfqxx

kyimao 发表于 2023-6-4 16:40
一样提示找不到xx社区，这列

根据代码，应该是将数据中第七列 "xx社区" 作为筛选条件，但出现了找不到该列的情况。可能有以下几种原因：

1. 数据表格中不存在名为“xx社区”的列。
2. 数据表格中“xx社区”这列存在，但列名与代码中写的不一致。
3. 数据表格中存在“xx社区”这列，但其实际内容并非所有行均为“xx社区”，导致筛选结果为空。

建议检查一下数据表格，看一下列名是否正确、实际内容是否如代码预期。另外，也可以通过打印出读取的原始数据看一下是否正确读入表格数据。