如何对CSV文件中的多列进行排序

import csv

def sort_csv_by_columns(csv_file, sort_columns):
    # 读取CSV文件
    with open(csv_file, 'r') as file:
        reader = csv.reader(file)
        data = list(reader)

    # 解析CSV文件
    header = data[0]
    rows = data[1:]

    # 选择排序列
    sort_indices = [header.index(column) for column in sort_columns]

    # 排序数据
    rows.sort(key=lambda row: [row[index] for index in sort_indices])

    # 应用排序结果
    sorted_data = [header] + rows

    # 写入CSV文件
    with open(csv_file, 'w', newline='') as file:
        writer = csv.writer(file)
        writer.writerows(sorted_data)

# 示例用法
csv_file = 'data.csv'
sort_columns = ['column1', 'column2']
sort_csv_by_columns(csv_file, sort_columns)