从.csv中读取前n行，并将列存储到列表中

import csv

def read_csv_first_n_rows(file_path, n):
    columns = []
    with open(file_path, newline='', encoding='utf-8') as csvfile:
        reader = csv.reader(csvfile)
        for i, row in enumerate(reader):
            if i >= n:
                break
            if i == 0:
                columns = row
            else:
                for j, value in enumerate(row):
                    if len(columns) <= j:
                        columns.append(value)
                    else:
                        columns[j] += ',' + value
    return columns

# 示例用法
file_path = 'example.csv'
n = 5
columns = read_csv_first_n_rows(file_path, n)
print(columns)