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

import csv

def read_csv_column(file_path, column_index, n):
    data = []  # 存储列数据的列表
    with open(file_path, 'r') as file:
        reader = csv.reader(file)
        for i, row in enumerate(reader):
            if i < n:  # 读取前n行
                data.append(row[column_index])  # 将指定列的数据添加到列表中
            else:
                break  # 达到指定行数后退出循环
    return data

file_path = 'example.csv'  # .csv文件路径
column_index = 0  # 要提取的列的索引，假设为第一列，索引从0开始
n = 5  # 要读取的行数

column_data = read_csv_column(file_path, column_index, n)
print(column_data)