将文件夹中不同文件类型的数据合并到pandas DataFrame中

import pandas as pd
import os

def merge_files(folder_path):
    file_list = os.listdir(folder_path)  # 获取文件夹中的所有文件
    df = pd.DataFrame()  # 创建一个空的DataFrame用于存储合并后的数据
    
    for file_name in file_list:
        file_path = os.path.join(folder_path, file_name)  # 获取文件的完整路径
        
        if file_name.endswith('.csv'):  # 如果文件是CSV格式
            data = pd.read_csv(file_path)  # 使用pandas的read_csv函数读取CSV文件
            df = df.append(data)  # 将读取的数据追加到DataFrame中
            
        elif file_name.endswith('.xlsx'):  # 如果文件是Excel格式
            data = pd.read_excel(file_path)  # 使用pandas的read_excel函数读取Excel文件
            df = df.append(data)  # 将读取的数据追加到DataFrame中
            
        # 可以根据需要添加其他文件类型的处理逻辑
        
    return df

folder_path = '文件夹路径'
merged_df = merge_files(folder_path)