要批量筛选CSV文件中第一行标题中包含关键词的字段,可以使用Python的`csv`模块来读取和写入CSV文件,并使用字符串的`find()`方法或正则表达式来检查字段是否包含关键词。以下是一个示例代码:
import csv
import os
def filter_csv_files(input_directory, output_directory, keyword):
for file_name in os.listdir(input_directory):
if file_name.endswith('.csv'):
input_path = os.path.join(input_directory, file_name)
output_path = os.path.join(output_directory, file_name)
with open(input_path, 'r', newline='') as input_file:
reader = csv.reader(input_file)
header = next(reader) # 读取第一行标题
# 筛选包含关键词的字段索引
keyword_fields = []
for i, field in enumerate(header):
if keyword in field:
keyword_fields.append(i)
with ope