python批量提取csv文件的特定列

可以使用Python中的pandas库来批量提取CSV文件的特定列。以下是一个示例代码：

import pandas as pd
import glob

# 获取所有CSV文件的文件名
csv_files = glob.glob('*.csv')

# 遍历每个CSV文件并提取特定列
for file in csv_files:
    # 读取CSV文件
    df = pd.read_csv(file)
    
    # 提取特定列，例如第2列和第4列
    specific_cols = [2, 4]
    df = df.iloc[:, specific_cols]
    
    # 将提取后的数据保存为新的CSV文件
    new_file_name = 'specific_cols_' + file
    df.to_csv(new_file_name, index=False)

这段代码将读取所有CSV文件，并提取第2列和第4列，然后将提取后的数据保存为新的CSV文件，文件名以“specific_cols_”开头。你可以根据需要修改代码来提取不同的列。