在Python中从目录中的多个CSV文件中拉出特定列

import os
import pandas as pd

def extract_columns_from_csv(directory, columns_to_extract):
    data_frames = []
    for filename in os.listdir(directory):
        if filename.endswith('.csv'):
            file_path = os.path.join(directory, filename)
            df = pd.read_csv(file_path, usecols=columns_to_extract)
            data_frames.append(df)
    combined_df = pd.concat(data_frames)
    return combined_df

directory_path = '路径/到/目录'
columns = ['列1', '列2', '列3']
result = extract_columns_from_csv(directory_path, columns)