有没有办法使用Python csv阅读器模块来提取一些列？

import csv

def extract_columns(csv_file, columns):
    with open(csv_file, 'r') as file:
        reader = csv.reader(file)
        header = next(reader)  # 获取csv文件的头部信息
        column_indices = [header.index(column) for column in columns]  # 获取指定列的索引
        extracted_data = [[row[index] for index in column_indices] for row in reader]  # 提取指定列的数据
    return extracted_data

# 示例用法
csv_file = 'data.csv'  # 替换为你的csv文件路径
columns = ['Column1', 'Column3']  # 替换为你要提取的列名
extracted_data = extract_columns(csv_file, columns)
print(extracted_data)