python按列索引提取資料夾內所有excel指定列彙總(範例程式碼)

# -*- coding:utf-8 –*- 
import os
import pandas as pd
 
# 輸入引數為excel表格所在目錄
def to_one_excel(dir):
    dfs = []
    # 遍歷檔案目錄，將所有表格表示為pandas中的DataFrame物件
    # for root_dir, sub_dir, files in os.walk(r'' + dir):     # 第一個為起始路徑，第二個為起始路徑下的資料夾，第三個是起始路徑下的檔案。
    for root_dir, sub_dir, files in os.walk(dir):     # 第一個為起始路徑，第二個為起始路徑下的資料夾，第三個是起始路徑下的檔案。
        for file in files:
            if file.endswith('xlsx'):
                # 構造絕對路徑
                file_name = os.path.join(root_dir, file)
                # df = pd.read_excel(file_name)
                df_1 = list(pd.read_excel(file_name, nrows=1))  # 讀取excel第一行資料並放進列表
                # excel第一行資料返回列表
                print(file_name)
                print(type(df_1))
                print(df_1)
                # 根據第一行列名獲取每個檔案中需要列的列索引，返回索引數值
                suo_yin_1 = df_1.index("人員")
                suo_yin_2 = df_1.index("效率")
                suo_yin_3 = df_1.index("成績")
                suo_yin_4 = df_1.index("產量")
                suo_yin_5 = df_1.index("節點")
                suo_yin_6 = df_1.index("備註")
                # 讀取檔案內容  usecols=[1, 3, 4] 讀取第1,3,4列
                df = pd.read_excel(file_name, usecols=[suo_yin_1, suo_yin_2, suo_yin_3, suo_yin_4, suo_yin_5, suo_yin_6], sheet_name='data')
                
                # pf = pd.read_excel('xxx.xls', usecols=[1, 3, 4], sheet_name='data')
                # print(pf)
                # 追加一列資料，將每個檔案的名字追加進該檔案的資料中，確定每條資料屬於哪個檔案
                excel_name = file.replace(".xlsx", "")      # 提取每個excel檔案的名稱，去掉.xlsx字尾
                df["檔名"] = excel_name       # 新建列名為「檔名」，列資料為excel檔名
                dfs.append(df)      # 將新建檔名列追加進彙總excel中
    # 行合併
    df_concated = pd.concat(dfs)
    # 構造輸出目錄的絕對路徑
    out_path = os.path.join(dir, 'res.xlsx')
    # 輸出到excel表格中，並刪除pandas預設的index列
    df_concated.to_excel(out_path, sheet_name='Sheet1', index=None)
# 呼叫並執行函數
to_one_excel(r'E:pypython3.7testtest96')