如何解析csv数据，将某些行显示为列

import csv

def parse_csv(filename):
    with open(filename, 'r') as file:
        reader = csv.reader(file)
        data = list(reader)
    return data

def transpose_rows_to_columns(data, rows_to_columns):
    transposed_data = {}
    for row in data:
        for column in rows_to_columns:
            if column not in transposed_data:
                transposed_data[column] = []
            if row[0] == column:
                transposed_data[column].extend(row[1:])
    return transposed_data

filename = 'data.csv'
data = parse_csv(filename)
rows_to_columns = ['A', 'B']  # 选择需要显示为列的行
transposed_data = transpose_rows_to_columns(data, rows_to_columns)

# 输出转置后的数据
for column, values in transposed_data.items():
    print(f"{column}: {', '.join(values)}")